Skip to content

Commit 13c1371

Browse files
author
Sean Morgan
committed
* Add framework_version to SKLearn estimator
1 parent 633fcf1 commit 13c1371

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)