Add the estimator base classes to API docs (#488)

laurenyu · web-flow · commit 11d3fcf431d7 · 2018-11-15T13:40:42.000-08:00
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -3,11 +3,12 @@ CHANGELOG
 =========
 
 1.14.3-dev
-=====
+======
 
 * bug-fix: Changes to use correct S3 bucket and time range for dataframes in TrainingJobAnalytics.
 * bug-fix: Local Mode: correctly handle the case where the model output folder doesn't exist yet
 * doc-fix: Fix typos in tensorflow serving documentation
+* doc-fix: Add estimator base classes to API docs
 
 1.14.2
 ======
diff --git a/doc/estimators.rst b/doc/estimators.rst
@@ -1,10 +1,22 @@
 Estimators
---------------------
+----------
 
 A high level interface for SageMaker training
 
+.. autoclass:: sagemaker.estimator.EstimatorBase
+    :members:
+    :undoc-members:
+    :show-inheritance:
+    :inherited-members:
+
 .. autoclass:: sagemaker.estimator.Estimator
     :members:
     :undoc-members:
     :show-inheritance:
     :inherited-members:
+
+.. autoclass:: sagemaker.estimator.Framework
+    :members:
+    :undoc-members:
+    :show-inheritance:
+    :inherited-members:
