Saving work

scikit_learn_ex/output/.gitignore

@@ -1,2 +0,0 @@
-*.pkl
-*.onnx

scikit_learn_ex/train.py

@@ -1,22 +0,0 @@
-from turtle import pd
-import joblib as jl
-
-from sklearn.datasets import load_iris
-from sklearn.model_selection import train_test_split
-from sklearn.ensemble import RandomForestClassifier
-
-def main():
-    # Use iris dataset
-    iris = load_iris()
-    X, y = iris.data, iris.target
-    X_train, X_test, y_train, y_test = train_test_split(X, y)
-    clr = RandomForestClassifier()
-    # Fit
-    clr.fit(X_train, y_train)
-    # Serialize the classifier to pickle file
-    jl.dump(clr, "./output/model.pkl", compress=9)
-
-if __name__ == "__main__":
-    print("Building iris model...")
-    main()
-    print("Model trained and dumped as pickle file.")

sklearn_ex/README.md

@@ -19,7 +19,7 @@ python convert_to_onnx.py ./output/model.pkl
 Make inferences
 
 ```
-
+python onnx_infer.py ./output/clr_model.onnx
 ```
 
 ## Model