Fix sample example documentation for python pkg (#61)

microsoft · Nov 30, 2018 · 39ebccb · 39ebccb
1 parent 1ea32a0
commit 39ebccb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/python/README.rst b/docs/python/README.rst
@@ -26,11 +26,11 @@ replaces *scikit-learn* to compute the predictions.
     # Train a model.
     from sklearn.datasets import load_iris
     from sklearn.model_selection import train_test_split
-    from sklearn.ensemble import RandomForest
+    from sklearn.ensemble import RandomForestClassifier
     iris = load_iris()
     X, y = iris.data, iris.target
     X_train, X_test, y_train, y_test = train_test_split(X, y)
-    clr = RandomForest()
+    clr = RandomForestClassifier()
     clr.fit(X_train, y_train)
 
     # Convert into ONNX format with onnxmltools