Create output_onnx_single_probability.py

onnx / sklearn-onnx

Convert scikit-learn models and pipelines to ONNX

Apache License 2.0

557 stars 104 forks source link

Closed ejosowitz closed 1 week ago

ejosowitz commented 1 week ago

Converts a RandomForestClassifier model to ONNX such that only a single (positive) probability is output

xadupre commented 1 week ago

Let's keep it that way. I'll add extra comments to improve the rendering through sphinx-gallery.