.. DO NOT EDIT. .. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY. .. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE: .. "auto_tutorial/plot_dbegin_options_list.py" .. LINE NUMBERS ARE GIVEN BELOW. .. only:: html .. note:: :class: sphx-glr-download-link-note :ref:`Go to the end ` to download the full example code .. rst-class:: sphx-glr-example-title .. _sphx_glr_auto_tutorial_plot_dbegin_options_list.py: Black list operators when converting ==================================== .. index:: black list, white list Some runtimes do not implement a runtime for every available operator in ONNX. The converter does not know that but it is possible to black some operators. Most of the converters do not change their behaviour, they fail if they use a black listed operator, a couple of them produces a different ONNX graph. GaussianMixture +++++++++++++++ The first converter to change its behaviour depending on a black list of operators is for model *GaussianMixture*. .. GENERATED FROM PYTHON SOURCE LINES 22-35 .. code-block:: default from timeit import timeit import numpy from onnxruntime import InferenceSession from sklearn.mixture import GaussianMixture from sklearn.datasets import load_iris from sklearn.model_selection import train_test_split from skl2onnx import to_onnx data = load_iris() X_train, X_test = train_test_split(data.data) model = GaussianMixture() model.fit(X_train) .. raw:: html

GaussianMixture()

In a Jupyter environment, please rerun this cell to show the HTML representation or trust the notebook.
On GitHub, the HTML representation is unable to render, please try loading this page with nbviewer.org.

.. GENERATED FROM PYTHON SOURCE LINES 36-38 Default conversion ++++++++++++++++++ .. GENERATED FROM PYTHON SOURCE LINES 38-54 .. code-block:: default model_onnx = to_onnx( model, X_train[:1].astype(numpy.float32), options={id(model): {"score_samples": True}}, target_opset=12, ) sess = InferenceSession( model_onnx.SerializeToString(), providers=["CPUExecutionProvider"] ) xt = X_test[:5].astype(numpy.float32) print(model.score_samples(xt)) print(sess.run(None, {"X": xt})[2]) .. rst-class:: sphx-glr-script-out .. code-block:: none [-1.87252497 -1.34625882 -3.3789712 -2.51747032 -2.11219732] [[-1.8725252] [-1.3462601] [-3.3789716] [-2.51747 ] [-2.112197 ]] .. GENERATED FROM PYTHON SOURCE LINES 55-61 Conversion without ReduceLogSumExp ++++++++++++++++++++++++++++++++++ Parameter *black_op* is used to tell the converter not to use this operator. Let's see what the converter produces in that case. .. GENERATED FROM PYTHON SOURCE LINES 61-77 .. code-block:: default model_onnx2 = to_onnx( model, X_train[:1].astype(numpy.float32), options={id(model): {"score_samples": True}}, black_op={"ReduceLogSumExp"}, target_opset=12, ) sess2 = InferenceSession( model_onnx2.SerializeToString(), providers=["CPUExecutionProvider"] ) xt = X_test[:5].astype(numpy.float32) print(model.score_samples(xt)) print(sess2.run(None, {"X": xt})[2]) .. rst-class:: sphx-glr-script-out .. code-block:: none [-1.87252497 -1.34625882 -3.3789712 -2.51747032 -2.11219732] [[-1.8725252] [-1.3462601] [-3.3789716] [-2.51747 ] [-2.112197 ]] .. GENERATED FROM PYTHON SOURCE LINES 78-80 Processing time +++++++++++++++ .. GENERATED FROM PYTHON SOURCE LINES 80-95 .. code-block:: default print( timeit( stmt="sess.run(None, {'X': xt})", number=10000, globals={"sess": sess, "xt": xt} ) ) print( timeit( stmt="sess2.run(None, {'X': xt})", number=10000, globals={"sess2": sess2, "xt": xt}, ) ) .. rst-class:: sphx-glr-script-out .. code-block:: none 0.5201874999997926 0.4130731999998716 .. GENERATED FROM PYTHON SOURCE LINES 96-97 The model using ReduceLogSumExp is much faster. .. GENERATED FROM PYTHON SOURCE LINES 99-106 If the converter cannot convert without... ++++++++++++++++++++++++++++++++++++++++++ Many converters do not consider the white and black lists of operators. If a converter fails to convert without using a blacklisted operator (or only whitelisted operators), *skl2onnx* raises an error. .. GENERATED FROM PYTHON SOURCE LINES 106-117 .. code-block:: default try: to_onnx( model, X_train[:1].astype(numpy.float32), options={id(model): {"score_samples": True}}, black_op={"ReduceLogSumExp", "Add"}, target_opset=12, ) except RuntimeError as e: print("Error:", e) .. rst-class:: sphx-glr-script-out .. code-block:: none Error: Operator 'Add' is black listed. .. rst-class:: sphx-glr-timing **Total running time of the script:** (0 minutes 1.122 seconds) .. _sphx_glr_download_auto_tutorial_plot_dbegin_options_list.py: .. only:: html .. container:: sphx-glr-footer sphx-glr-footer-example .. container:: sphx-glr-download sphx-glr-download-python :download:`Download Python source code: plot_dbegin_options_list.py ` .. container:: sphx-glr-download sphx-glr-download-jupyter :download:`Download Jupyter notebook: plot_dbegin_options_list.ipynb ` .. only:: html .. rst-class:: sphx-glr-signature `Gallery generated by Sphinx-Gallery `_