From 83c40deac50529b51c5b42f8141945ad18477ab7 Mon Sep 17 00:00:00 2001 From: Rui Wang Date: Wed, 31 Jul 2019 14:48:05 -0700 Subject: [PATCH] fix: update sklearn document to include 3p dependency installation --- doc/using_sklearn.rst | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/doc/using_sklearn.rst b/doc/using_sklearn.rst index b72d2b4356..a08d2223f5 100644 --- a/doc/using_sklearn.rst +++ b/doc/using_sklearn.rst @@ -97,6 +97,16 @@ inadvertently run your training code at the wrong point in execution. For more on training environment variables, please visit https://github.com/aws/sagemaker-containers. +Using third-party libraries +^^^^^^^^^^^^^^^^^^^^^^^^^^^ + +When running your training script on SageMaker, it will have access to some pre-installed third-party libraries including ``scikit-learn``, ``numpy``, and ``pandas``. +For more information on the runtime environment, including specific package versions, see `SageMaker Scikit-learn Docker containers `__. + +If there are other packages you want to use with your script, you can include a ``requirements.txte` file in the same directory as your training script to install other dependencies at runtime. +A ``requirements.txt`` file is a text file that contains a list of items that are installed by using ``pip install``. You can also specify the version of an item to install. +For information about the format of a ``requirements.txt`` file, see `Requirements Files `__ in the pip documentation. + Running a Scikit-learn training script in SageMaker ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~