Skip to content

Commit 5f14219

Browse files
authored
fix: update sklearn document to include 3p dependency installation (#960)
1 parent b8937d7 commit 5f14219

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

doc/using_sklearn.rst

+10
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,16 @@ inadvertently run your training code at the wrong point in execution.
9797

9898
For more on training environment variables, please visit https://github.com/aws/sagemaker-containers.
9999

100+
Using third-party libraries
101+
^^^^^^^^^^^^^^^^^^^^^^^^^^^
102+
103+
When running your training script on SageMaker, it will have access to some pre-installed third-party libraries including ``scikit-learn``, ``numpy``, and ``pandas``.
104+
For more information on the runtime environment, including specific package versions, see `SageMaker Scikit-learn Docker containers <https://github.com/aws/sagemaker-scikit-learn-container>`__.
105+
106+
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.
107+
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.
108+
For information about the format of a ``requirements.txt`` file, see `Requirements Files <https://pip.pypa.io/en/stable/user_guide/#requirements-files>`__ in the pip documentation.
109+
100110
Running a Scikit-learn training script in SageMaker
101111
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
102112

0 commit comments

Comments
 (0)