Skip to content

Commit ef9d461

Browse files
seanpmorganSean Morgan
andauthored
* Add framework_version to SKLearn estimator (#1716)
Co-authored-by: Sean Morgan <[email protected]>
1 parent 4fccde7 commit ef9d461

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sagemaker_processing/scikit_learn_data_processing_and_model_evaluation/scikit_learn_data_processing_and_model_evaluation.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -254,6 +254,7 @@
254254
"\n",
255255
"sklearn = SKLearn(\n",
256256
" entry_point='train.py',\n",
257+
" framework_version='0.20.0',\n",
257258
" train_instance_type=\"ml.m5.xlarge\",\n",
258259
" role=role)"
259260
]
@@ -564,4 +565,4 @@
564565
},
565566
"nbformat": 4,
566567
"nbformat_minor": 4
567-
}
568+
}

0 commit comments

Comments
 (0)