mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-18 21:21:17 +00:00
Merge branch 'master' of https://github.com/Microsoft/onnxruntime into kezhan/remove_const_cast
This commit is contained in:
commit
fbd7b9b8f0
1 changed files with 2 additions and 2 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in a new issue