From 39ebccbc8b2c0285dc5fe5d064bf99bf68b52aa1 Mon Sep 17 00:00:00 2001
From: Pranav Sharma <emailpranav@gmail.com>
Date: Thu, 29 Nov 2018 17:50:12 -0800
Subject: [PATCH] Fix sample example documentation for python pkg (#61)

---
 docs/python/README.rst | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/python/README.rst b/docs/python/README.rst
index 97b1a3b34f..bf84a2b32c 100644
--- 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