You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Drew Bednar 94c218ff1d Saving sklearn example 3 years ago
..
img Saving sklearn example 3 years ago
output Saving sklearn example 3 years ago
README.md Saving sklearn example 3 years ago
convert_to_onnx.py Saving sklearn example 3 years ago
onnx_infer.py Saving sklearn example 3 years ago
train.py Saving sklearn example 3 years ago

README.md

sklearn example

This is a simple demo showing the training of Randmon Forest Classifier, serializing it as a pickle file, converting it to an Onnx model, then making inferences using the Onnx model.

Usage

Train the model

python train.py

Convert to Onnx


Make inferences


Model

Model Graph