diff --git a/doc/index.rst b/doc/index.rst index db74369e73..5597b8dafc 100644 --- a/doc/index.rst +++ b/doc/index.rst @@ -48,6 +48,15 @@ A managed environment for PyTorch training and hosting on Amazon SageMaker sagemaker.pytorch +Scikit-learn +------------ +A managed environment for Scikit-learn training and hosting on Amazon SageMaker + +.. toctree:: + :maxdepth: 2 + + sagemaker.sklearn + Chainer ------- A managed environment for Chainer training and hosting on Amazon SageMaker @@ -80,6 +89,7 @@ Amazon provides implementations of some common machine learning algortithms opti factorization_machines lda ntm + randomcutforest Workflows --------- diff --git a/doc/sagemaker.sklearn.rst b/doc/sagemaker.sklearn.rst new file mode 100644 index 0000000000..c4019ea0c1 --- /dev/null +++ b/doc/sagemaker.sklearn.rst @@ -0,0 +1,26 @@ +Scikit-learn +======= + +Scikit-learn Estimator +---------------------- + +.. autoclass:: sagemaker.sklearn.estimator.SKLearn + :members: + :undoc-members: + :show-inheritance: + +Scikit-learn Model +------------------ + +.. autoclass:: sagemaker.sklearn.model.SKLearnModel + :members: + :undoc-members: + :show-inheritance: + +Scikit-learn Predictor +---------------------- + +.. autoclass:: sagemaker.sklearn.model.SKLearnPredictor + :members: + :undoc-members: + :show-inheritance: