.. 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-36 .. code-block:: Python 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 37-39 Default conversion ++++++++++++++++++ .. GENERATED FROM PYTHON SOURCE LINES 39-55 .. code-block:: Python 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.90576222 -0.96229758 -2.40249961 -4.2508906 -5.73341576] [[-1.9057629] [-0.9622972] [-2.4025004] [-4.2508907] [-5.7334156]] .. GENERATED FROM PYTHON SOURCE LINES 56-62 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 62-78 .. code-block:: Python 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.90576222 -0.96229758 -2.40249961 -4.2508906 -5.73341576] [[-1.9057629] [-0.9622972] [-2.4025004] [-4.2508907] [-5.7334156]] .. GENERATED FROM PYTHON SOURCE LINES 79-81 Processing time +++++++++++++++ .. GENERATED FROM PYTHON SOURCE LINES 81-96 .. code-block:: Python 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.22418856500007678 0.28478742000152124 .. GENERATED FROM PYTHON SOURCE LINES 97-98 The model using ReduceLogSumExp is much faster. .. GENERATED FROM PYTHON SOURCE LINES 100-107 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 107-118 .. code-block:: Python 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 0.618 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-jupyter :download:`Download Jupyter notebook: plot_dbegin_options_list.ipynb ` .. 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-zip :download:`Download zipped: plot_dbegin_options_list.zip ` .. only:: html .. rst-class:: sphx-glr-signature `Gallery generated by Sphinx-Gallery `_