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 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~