learning
/
learn_onnx
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
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.
master
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'master'
${ noResults }
learn_onnx/sklearn_ex
History
Drew Bednar 14a3a6f182 Saving work 3 years ago
..
img Saving sklearn example 3 years ago
output Saving sklearn example 3 years ago
README.md Saving work 3 years ago
convert_to_onnx.py Saving sklearn example 3 years ago
onnx_infer.py Adding basic onnx inference on 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 Random 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

python convert_to_onnx.py ./output/model.pkl

Make inferences

python onnx_infer.py ./output/clr_model.onnx

Model

Model Graph