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.

443 B

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