2019-12-21 01:11:46 +00:00
<!DOCTYPE html>
2020-08-13 02:12:50 +00:00
< html lang = "en" >
2019-12-21 01:11:46 +00:00
< head >
< meta charset = "utf-8" / >
2023-01-21 18:34:16 +00:00
< meta name = "viewport" content = "width=device-width, initial-scale=1.0" / > < meta name = "generator" content = "Docutils 0.17.1: http://docutils.sourceforge.net/" / >
2022-01-04 17:32:40 +00:00
2022-12-09 23:55:42 +00:00
<!-- Google tag (gtag.js) -->
< script async src = "https://www.googletagmanager.com/gtag/js?id=UA-156955408-1" > < / script >
< script >
window.dataLayer = window.dataLayer || [];
function gtag(){dataLayer.push(arguments);}
gtag('js', new Date());
gtag('config', 'UA-156955408-1');
< / script >
< title > Train, convert and predict with ONNX Runtime — ONNX Runtime 1.14.0 documentation< / title >
2022-01-04 17:32:40 +00:00
< link rel = "stylesheet" type = "text/css" href = "../static/pygments.css" / >
< link rel = "stylesheet" type = "text/css" href = "../static/alabaster.css" / >
2021-02-11 00:34:31 +00:00
< link rel = "stylesheet" type = "text/css" href = "../static/graphviz.css" / >
2022-01-04 17:32:40 +00:00
< link rel = "stylesheet" type = "text/css" href = "../static/sg_gallery.css" / >
< link rel = "stylesheet" type = "text/css" href = "../static/sg_gallery-binder.css" / >
< link rel = "stylesheet" type = "text/css" href = "../static/sg_gallery-dataframe.css" / >
< link rel = "stylesheet" type = "text/css" href = "../static/sg_gallery-rendered-html.css" / >
< script data-url_root = "../" id = "documentation_options" src = "../static/documentation_options.js" > < / script >
2023-01-21 18:34:16 +00:00
< script src = "../static/jquery.js" > < / script >
< script src = "../static/underscore.js" > < / script >
< script src = "../static/_sphinx_javascript_frameworks_compat.js" > < / script >
2021-02-11 00:34:31 +00:00
< script src = "../static/doctools.js" > < / script >
2022-12-09 23:55:42 +00:00
< script src = "../static/sphinx_highlight.js" > < / script >
2019-12-21 01:11:46 +00:00
< link rel = "index" title = "Index" href = "../genindex.html" / >
< link rel = "search" title = "Search" href = "../search.html" / >
< link rel = "prev" title = "Common errors with onnxruntime" href = "plot_common_errors.html" / >
2021-02-11 00:34:31 +00:00
< link rel = "stylesheet" href = "../static/custom.css" type = "text/css" / >
2019-12-21 01:11:46 +00:00
< meta name = "viewport" content = "width=device-width, initial-scale=0.9, maximum-scale=0.9" / >
< / head > < body >
< div class = "document" >
< div class = "documentwrapper" >
< div class = "bodywrapper" >
< div class = "body" role = "main" >
< div class = "sphx-glr-download-link-note admonition note" >
< p class = "admonition-title" > Note< / p >
2021-03-09 18:21:38 +00:00
< p > Click < a class = "reference internal" href = "#sphx-glr-download-auto-examples-plot-train-convert-predict-py" > < span class = "std std-ref" > here< / span > < / a >
to download the full example code< / p >
2019-12-21 01:11:46 +00:00
< / div >
2022-01-04 17:32:40 +00:00
< section class = "sphx-glr-example-title" id = "train-convert-and-predict-with-onnx-runtime" >
2022-08-03 17:24:42 +00:00
< span id = "l-logreg-example" > < / span > < span id = "sphx-glr-auto-examples-plot-train-convert-predict-py" > < / span > < h1 > Train, convert and predict with ONNX Runtime< a class = "headerlink" href = "#train-convert-and-predict-with-onnx-runtime" title = "Permalink to this heading" > ¶< / a > < / h1 >
2019-12-21 01:11:46 +00:00
< p > This example demonstrates an end to end scenario
starting with the training of a machine learned model
to its use in its converted from.< / p >
2023-01-21 18:34:16 +00:00
< div class = "contents local topic" id = "contents" >
2019-12-21 01:11:46 +00:00
< ul class = "simple" >
< li > < p > < a class = "reference internal" href = "#train-a-logistic-regression" id = "id1" > Train a logistic regression< / a > < / p > < / li >
< li > < p > < a class = "reference internal" href = "#conversion-to-onnx-format" id = "id2" > Conversion to ONNX format< / a > < / p > < / li >
< li > < p > < a class = "reference internal" href = "#probabilities" id = "id3" > Probabilities< / a > < / p > < / li >
< li > < p > < a class = "reference internal" href = "#benchmark-with-randomforest" id = "id4" > Benchmark with RandomForest< / a > < / p > < / li >
< / ul >
2023-01-21 18:34:16 +00:00
< / div >
2022-01-04 17:32:40 +00:00
< section id = "train-a-logistic-regression" >
2023-01-21 18:34:16 +00:00
< h2 > < a class = "toc-backref" href = "#id1" > Train a logistic regression< / a > < a class = "headerlink" href = "#train-a-logistic-regression" title = "Permalink to this heading" > ¶< / a > < / h2 >
2019-12-21 01:11:46 +00:00
< p > The first step consists in retrieving the iris datset.< / p >
2022-12-09 23:55:42 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "kn" > from< / span > < span class = "nn" > sklearn.datasets< / span > < span class = "kn" > import< / span > < span class = "n" > load_iris< / span >
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "n" > iris< / span > < span class = "o" > =< / span > < span class = "n" > load_iris< / span > < span class = "p" > ()< / span >
< span class = "n" > X< / span > < span class = "p" > ,< / span > < span class = "n" > y< / span > < span class = "o" > =< / span > < span class = "n" > iris< / span > < span class = "o" > .< / span > < span class = "n" > data< / span > < span class = "p" > ,< / span > < span class = "n" > iris< / span > < span class = "o" > .< / span > < span class = "n" > target< / span >
2022-12-09 23:55:42 +00:00
< span class = "kn" > from< / span > < span class = "nn" > sklearn.model_selection< / span > < span class = "kn" > import< / span > < span class = "n" > train_test_split< / span >
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "n" > X_train< / span > < span class = "p" > ,< / span > < span class = "n" > X_test< / span > < span class = "p" > ,< / span > < span class = "n" > y_train< / span > < span class = "p" > ,< / span > < span class = "n" > y_test< / span > < span class = "o" > =< / span > < span class = "n" > train_test_split< / span > < span class = "p" > (< / span > < span class = "n" > X< / span > < span class = "p" > ,< / span > < span class = "n" > y< / span > < span class = "p" > )< / span >
< / pre > < / div >
< / div >
< p > Then we fit a model.< / p >
2022-12-09 23:55:42 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "kn" > from< / span > < span class = "nn" > sklearn.linear_model< / span > < span class = "kn" > import< / span > < span class = "n" > LogisticRegression< / span >
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "n" > clr< / span > < span class = "o" > =< / span > < span class = "n" > LogisticRegression< / span > < span class = "p" > ()< / span >
< span class = "n" > clr< / span > < span class = "o" > .< / span > < span class = "n" > fit< / span > < span class = "p" > (< / span > < span class = "n" > X_train< / span > < span class = "p" > ,< / span > < span class = "n" > y_train< / span > < span class = "p" > )< / span >
< / pre > < / div >
< / div >
2023-01-20 00:19:31 +00:00
< div class = "sphx-glr-script-out highlight-none notranslate" > < div class = "highlight" > < pre > < span > < / span > /home/runner/.local/lib/python3.10/site-packages/sklearn/linear_model/_logistic.py:444: ConvergenceWarning: lbfgs failed to converge (status=1):
STOP: TOTAL NO. of ITERATIONS REACHED LIMIT.
Increase the number of iterations (max_iter) or scale the data as shown in:
https://scikit-learn.org/stable/modules/preprocessing.html
Please also refer to the documentation for alternative solver options:
https://scikit-learn.org/stable/modules/linear_model.html#logistic-regression
n_iter_i = _check_optimize_result(
< / pre > < / div >
< / div >
2022-08-03 17:24:42 +00:00
< div class = "output_subarea output_html rendered_html output_result" >
< style > # sk-container-id-2 { color : black ; background-color : white ; } # sk-container-id-2 pre { padding : 0 ; } # sk-container-id-2 div . sk-toggleable { background-color : white ; } # sk-container-id-2 label . sk-toggleable__label { cursor : pointer ; display : block ; width : 100 % ; margin-bottom : 0 ; padding : 0.3 em ; box-sizing : border-box ; text-align : center ; } # sk-container-id-2 label . sk-toggleable__label-arrow : before { content : "▸" ; float : left ; margin-right : 0.25 em ; color : #696969 ; } # sk-container-id-2 label . sk-toggleable__label-arrow : hover : before { color : black ; } # sk-container-id-2 div . sk-estimator : hover label . sk-toggleable__label-arrow : before { color : black ; } # sk-container-id-2 div . sk-toggleable__content { max-height : 0 ; max-width : 0 ; overflow : hidden ; text-align : left ; background-color : #f0f8ff ; } # sk-container-id-2 div . sk-toggleable__content pre { margin : 0.2 em ; color : black ; border-radius : 0.25 em ; background-color : #f0f8ff ; } # sk-container-id-2 input . sk-toggleable__control : checked ~ div . sk-toggleable__content { max-height : 200 px ; max-width : 100 % ; overflow : auto ; } # sk-container-id-2 input . sk-toggleable__control : checked ~ label . sk-toggleable__label-arrow : before { content : "▾" ; } # sk-container-id-2 div . sk-estimator input . sk-toggleable__control : checked ~ label . sk-toggleable__label { background-color : #d4ebff ; } # sk-container-id-2 div . sk-label input . sk-toggleable__control : checked ~ label . sk-toggleable__label { background-color : #d4ebff ; } # sk-container-id-2 input . sk-hidden--visually { border : 0 ; clip : rect ( 1 px 1 px 1 px 1 px ) ; clip : rect ( 1 px , 1 px , 1 px , 1 px ) ; height : 1 px ; margin : -1 px ; overflow : hidden ; padding : 0 ; position : absolute ; width : 1 px ; } # sk-container-id-2 div . sk-estimator { font-family : monospace ; background-color : #f0f8ff ; border : 1 px dotted black ; border-radius : 0.25 em ; box-sizing : border-box ; margin-bottom : 0.5 em ; } # sk-container-id-2 div . sk-estimator : hover { background-color : #d4ebff ; } # sk-container-id-2 div . sk-parallel-item :: after { content : "" ; width : 100 % ; border-bottom : 1 px solid gray ; flex-grow : 1 ; } # sk-container-id-2 div . sk-label : hover label . sk-toggleable__label { background-color : #d4ebff ; } # sk-container-id-2 div . sk-serial :: before { content : "" ; position : absolute ; border-left : 1 px solid gray ; box-sizing : border-box ; top : 0 ; bottom : 0 ; left : 50 % ; z-index : 0 ; } # sk-container-id-2 div . sk-serial { display : flex ; flex-direction : column ; align-items : center ; background-color : white ; padding-right : 0.2 em ; padding-left : 0.2 em ; position : relative ; } # sk-container-id-2 div . sk-item { position : relative ; z-index : 1 ; } # sk-container-id-2 div . sk-parallel { display : flex ; align-items : stretch ; justify-content : center ; background-color : white ; position : relative ; } # sk-container-id-2 div . sk-item :: before , # sk-container-id-2 div . sk-parallel-item :: before { content : "" ; position : absolute ; border-left : 1 px solid gray ; box-sizing : border-box ; top : 0 ; bottom : 0 ; left : 50 % ; z-index : -1 ; } # sk-container-id-2 div . sk-parallel-item { display : flex ; flex-direction : column ; z-index : 1 ; position : relative ; background-color : white ; } # sk-container-id-2 div . sk-parallel-item : first-child :: after { align-self : flex-end ; width : 50 % ; } # sk-container-id-2 div . sk-parallel-item : last-child :: after { align-self : flex-start ; width : 50 % ; } # sk-container-id-2 div . sk-parallel-item : only-child :: after { width : 0 ; } # sk-container-id-2 div . sk-dashed-wrapped { border : 1 px dashed gray ; margin : 0 0.4 em 0.5 em 0.4 em ; box-sizing : border-box ; padding-bottom : 0.4 em ; background-color : white ; } # sk-container-id-2 div . sk-label label { font-family : monospace ; font-weight : bold ; display : inline-block ; line-height : 1.2 em ; } # sk-container-id-2 div . sk-label-container { text-align : center ; } # sk-container-id-2 div . sk-container { /* jupyter's `normalize.less` sets `[hidden] { display: none; }` but bootstrap.min.css set `[hidden] { display: none !important; }` so we also need the `!important` here to be able to override the default hidden behavior on the sphinx rendered scikit-learn.org. See: https://github.com/scikit-learn/scikit-learn/issues/21755 */ display : inline-block !important ; position : relative ; } # sk-container-id-2 div . sk-text-repr-fallback { display : none ; } < / style > < div id = "sk-container-id-2" class = "sk-top-container" > < div class = "sk-text-
< / div >
< br / >
< br / > < p > We compute the prediction on the test set
2019-12-21 01:11:46 +00:00
and we show the confusion matrix.< / p >
2022-12-09 23:55:42 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "kn" > from< / span > < span class = "nn" > sklearn.metrics< / span > < span class = "kn" > import< / span > < span class = "n" > confusion_matrix< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > pred< / span > < span class = "o" > =< / span > < span class = "n" > clr< / span > < span class = "o" > .< / span > < span class = "n" > predict< / span > < span class = "p" > (< / span > < span class = "n" > X_test< / span > < span class = "p" > )< / span >
2020-08-13 02:12:50 +00:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "n" > confusion_matrix< / span > < span class = "p" > (< / span > < span class = "n" > y_test< / span > < span class = "p" > ,< / span > < span class = "n" > pred< / span > < span class = "p" > ))< / span >
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
2023-01-20 00:19:31 +00:00
< div class = "sphx-glr-script-out highlight-none notranslate" > < div class = "highlight" > < pre > < span > < / span > [[11 0 0]
2023-01-21 18:34:16 +00:00
[ 0 13 2]
[ 0 1 11]]
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
2022-01-04 17:32:40 +00:00
< / section >
< section id = "conversion-to-onnx-format" >
2023-01-21 18:34:16 +00:00
< h2 > < a class = "toc-backref" href = "#id2" > Conversion to ONNX format< / a > < a class = "headerlink" href = "#conversion-to-onnx-format" title = "Permalink to this heading" > ¶< / a > < / h2 >
2019-12-21 01:11:46 +00:00
< p > We use module
< a class = "reference external" href = "https://github.com/onnx/sklearn-onnx" > sklearn-onnx< / a >
to convert the model into ONNX format.< / p >
2022-12-09 23:55:42 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "kn" > from< / span > < span class = "nn" > skl2onnx< / span > < span class = "kn" > import< / span > < span class = "n" > convert_sklearn< / span >
< span class = "kn" > from< / span > < span class = "nn" > skl2onnx.common.data_types< / span > < span class = "kn" > import< / span > < span class = "n" > FloatTensorType< / span >
2019-12-21 01:11:46 +00:00
2022-08-03 17:24:42 +00:00
< span class = "n" > initial_type< / span > < span class = "o" > =< / span > < span class = "p" > [(< / span > < span class = "s2" > " float_input" < / span > < span class = "p" > ,< / span > < span class = "n" > FloatTensorType< / span > < span class = "p" > ([< / span > < span class = "kc" > None< / span > < span class = "p" > ,< / span > < span class = "mi" > 4< / span > < span class = "p" > ]))]< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > onx< / span > < span class = "o" > =< / span > < span class = "n" > convert_sklearn< / span > < span class = "p" > (< / span > < span class = "n" > clr< / span > < span class = "p" > ,< / span > < span class = "n" > initial_types< / span > < span class = "o" > =< / span > < span class = "n" > initial_type< / span > < span class = "p" > )< / span >
< span class = "k" > with< / span > < span class = "nb" > open< / span > < span class = "p" > (< / span > < span class = "s2" > " logreg_iris.onnx" < / span > < span class = "p" > ,< / span > < span class = "s2" > " wb" < / span > < span class = "p" > )< / span > < span class = "k" > as< / span > < span class = "n" > f< / span > < span class = "p" > :< / span >
< span class = "n" > f< / span > < span class = "o" > .< / span > < span class = "n" > write< / span > < span class = "p" > (< / span > < span class = "n" > onx< / span > < span class = "o" > .< / span > < span class = "n" > SerializeToString< / span > < span class = "p" > ())< / span >
< / pre > < / div >
< / div >
< p > We load the model with ONNX Runtime and look at
its input and output.< / p >
2020-08-13 02:12:50 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "kn" > import< / span > < span class = "nn" > onnxruntime< / span > < span class = "k" > as< / span > < span class = "nn" > rt< / span >
2022-08-03 17:24:42 +00:00
2022-01-04 17:32:40 +00:00
< span class = "n" > sess< / span > < span class = "o" > =< / span > < span class = "n" > rt< / span > < span class = "o" > .< / span > < span class = "n" > InferenceSession< / span > < span class = "p" > (< / span > < span class = "s2" > " logreg_iris.onnx" < / span > < span class = "p" > ,< / span > < span class = "n" > providers< / span > < span class = "o" > =< / span > < span class = "n" > rt< / span > < span class = "o" > .< / span > < span class = "n" > get_available_providers< / span > < span class = "p" > ())< / span >
2019-12-21 01:11:46 +00:00
2022-08-03 17:24:42 +00:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s2" > " input name=' < / span > < span class = "si" > {}< / span > < span class = "s2" > ' and shape=< / span > < span class = "si" > {}< / span > < span class = "s2" > " < / span > < span class = "o" > .< / span > < span class = "n" > format< / span > < span class = "p" > (< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > get_inputs< / span > < span class = "p" > ()[< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span > < span class = "o" > .< / span > < span class = "n" > name< / span > < span class = "p" > ,< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > get_inputs< / span > < span class = "p" > ()[< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span > < span class = "o" > .< / span > < span class = "n" > shape< / span > < span class = "p" > ))< / span >
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s2" > " output name=' < / span > < span class = "si" > {}< / span > < span class = "s2" > ' and shape=< / span > < span class = "si" > {}< / span > < span class = "s2" > " < / span > < span class = "o" > .< / span > < span class = "n" > format< / span > < span class = "p" > (< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > get_outputs< / span > < span class = "p" > ()[< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span > < span class = "o" > .< / span > < span class = "n" > name< / span > < span class = "p" > ,< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > get_outputs< / span > < span class = "p" > ()[< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span > < span class = "o" > .< / span > < span class = "n" > shape< / span > < span class = "p" > ))< / span >
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
< div class = "sphx-glr-script-out highlight-none notranslate" > < div class = "highlight" > < pre > < span > < / span > input name=' float_input' and shape=[None, 4]
2020-08-13 02:12:50 +00:00
output name=' output_label' and shape=[None]
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
< p > We compute the predictions.< / p >
2020-08-13 02:12:50 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "n" > input_name< / span > < span class = "o" > =< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > get_inputs< / span > < span class = "p" > ()[< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span > < span class = "o" > .< / span > < span class = "n" > name< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > label_name< / span > < span class = "o" > =< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > get_outputs< / span > < span class = "p" > ()[< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span > < span class = "o" > .< / span > < span class = "n" > name< / span >
< span class = "kn" > import< / span > < span class = "nn" > numpy< / span >
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "n" > pred_onx< / span > < span class = "o" > =< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > run< / span > < span class = "p" > ([< / span > < span class = "n" > label_name< / span > < span class = "p" > ],< / span > < span class = "p" > {< / span > < span class = "n" > input_name< / span > < span class = "p" > :< / span > < span class = "n" > X_test< / span > < span class = "o" > .< / span > < span class = "n" > astype< / span > < span class = "p" > (< / span > < span class = "n" > numpy< / span > < span class = "o" > .< / span > < span class = "n" > float32< / span > < span class = "p" > )})[< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span >
2020-08-13 02:12:50 +00:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "n" > confusion_matrix< / span > < span class = "p" > (< / span > < span class = "n" > pred< / span > < span class = "p" > ,< / span > < span class = "n" > pred_onx< / span > < span class = "p" > ))< / span >
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
2023-01-20 00:19:31 +00:00
< div class = "sphx-glr-script-out highlight-none notranslate" > < div class = "highlight" > < pre > < span > < / span > [[11 0 0]
2023-01-21 18:34:16 +00:00
[ 0 14 0]
[ 0 0 13]]
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
< p > The prediction are perfectly identical.< / p >
2022-01-04 17:32:40 +00:00
< / section >
< section id = "probabilities" >
2023-01-21 18:34:16 +00:00
< h2 > < a class = "toc-backref" href = "#id3" > Probabilities< / a > < a class = "headerlink" href = "#probabilities" title = "Permalink to this heading" > ¶< / a > < / h2 >
2019-12-21 01:11:46 +00:00
< p > Probabilities are needed to compute other
relevant metrics such as the ROC Curve.
Let’
scikit-learn.< / p >
2020-08-13 02:12:50 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "n" > prob_sklearn< / span > < span class = "o" > =< / span > < span class = "n" > clr< / span > < span class = "o" > .< / span > < span class = "n" > predict_proba< / span > < span class = "p" > (< / span > < span class = "n" > X_test< / span > < span class = "p" > )< / span >
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "n" > prob_sklearn< / span > < span class = "p" > [:< / span > < span class = "mi" > 3< / span > < span class = "p" > ])< / span >
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
2023-01-21 18:34:16 +00:00
< div class = "sphx-glr-script-out highlight-none notranslate" > < div class = "highlight" > < pre > < span > < / span > [[1.29030347e-03 5.19977113e-01 4.78732584e-01]
[7.90572203e-04 3.26327127e-01 6.72882300e-01]
[9.65026010e-01 3.49738628e-02 1.26697184e-07]]
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
< p > And then with ONNX Runtime.
The probabilies appear to be< / p >
2020-08-13 02:12:50 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "n" > prob_name< / span > < span class = "o" > =< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > get_outputs< / span > < span class = "p" > ()[< / span > < span class = "mi" > 1< / span > < span class = "p" > ]< / span > < span class = "o" > .< / span > < span class = "n" > name< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > prob_rt< / span > < span class = "o" > =< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > run< / span > < span class = "p" > ([< / span > < span class = "n" > prob_name< / span > < span class = "p" > ],< / span > < span class = "p" > {< / span > < span class = "n" > input_name< / span > < span class = "p" > :< / span > < span class = "n" > X_test< / span > < span class = "o" > .< / span > < span class = "n" > astype< / span > < span class = "p" > (< / span > < span class = "n" > numpy< / span > < span class = "o" > .< / span > < span class = "n" > float32< / span > < span class = "p" > )})[< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span >
< span class = "kn" > import< / span > < span class = "nn" > pprint< / span >
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "n" > pprint< / span > < span class = "o" > .< / span > < span class = "n" > pprint< / span > < span class = "p" > (< / span > < span class = "n" > prob_rt< / span > < span class = "p" > [< / span > < span class = "mi" > 0< / span > < span class = "p" > :< / span > < span class = "mi" > 3< / span > < span class = "p" > ])< / span >
< / pre > < / div >
< / div >
2023-01-21 18:34:16 +00:00
< div class = "sphx-glr-script-out highlight-none notranslate" > < div class = "highlight" > < pre > < span > < / span > [{0: 0.0012903024908155203, 1: 0.5199770927429199, 2: 0.4787325859069824},
{0: 0.0007905727834440768, 1: 0.32632729411125183, 2: 0.6728821396827698},
{0: 0.9650260210037231, 1: 0.034973856061697006, 2: 1.2669713100876834e-07}]
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
< p > Let’ < / p >
2022-12-09 23:55:42 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "kn" > from< / span > < span class = "nn" > timeit< / span > < span class = "kn" > import< / span > < span class = "n" > Timer< / span >
2019-12-21 01:11:46 +00:00
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "k" > def< / span > < span class = "nf" > speed< / span > < span class = "p" > (< / span > < span class = "n" > inst< / span > < span class = "p" > ,< / span > < span class = "n" > number< / span > < span class = "o" > =< / span > < span class = "mi" > 10< / span > < span class = "p" > ,< / span > < span class = "n" > repeat< / span > < span class = "o" > =< / span > < span class = "mi" > 20< / span > < span class = "p" > ):< / span >
< span class = "n" > timer< / span > < span class = "o" > =< / span > < span class = "n" > Timer< / span > < span class = "p" > (< / span > < span class = "n" > inst< / span > < span class = "p" > ,< / span > < span class = "nb" > globals< / span > < span class = "o" > =< / span > < span class = "nb" > globals< / span > < span class = "p" > ())< / span >
< span class = "n" > raw< / span > < span class = "o" > =< / span > < span class = "n" > numpy< / span > < span class = "o" > .< / span > < span class = "n" > array< / span > < span class = "p" > (< / span > < span class = "n" > timer< / span > < span class = "o" > .< / span > < span class = "n" > repeat< / span > < span class = "p" > (< / span > < span class = "n" > repeat< / span > < span class = "p" > ,< / span > < span class = "n" > number< / span > < span class = "o" > =< / span > < span class = "n" > number< / span > < span class = "p" > ))< / span >
< span class = "n" > ave< / span > < span class = "o" > =< / span > < span class = "n" > raw< / span > < span class = "o" > .< / span > < span class = "n" > sum< / span > < span class = "p" > ()< / span > < span class = "o" > /< / span > < span class = "nb" > len< / span > < span class = "p" > (< / span > < span class = "n" > raw< / span > < span class = "p" > )< / span > < span class = "o" > /< / span > < span class = "n" > number< / span >
< span class = "n" > mi< / span > < span class = "p" > ,< / span > < span class = "n" > ma< / span > < span class = "o" > =< / span > < span class = "n" > raw< / span > < span class = "o" > .< / span > < span class = "n" > min< / span > < span class = "p" > ()< / span > < span class = "o" > /< / span > < span class = "n" > number< / span > < span class = "p" > ,< / span > < span class = "n" > raw< / span > < span class = "o" > .< / span > < span class = "n" > max< / span > < span class = "p" > ()< / span > < span class = "o" > /< / span > < span class = "n" > number< / span >
2020-08-13 02:12:50 +00:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s2" > " Average < / span > < span class = "si" > %1.3g< / span > < span class = "s2" > min=< / span > < span class = "si" > %1.3g< / span > < span class = "s2" > max=< / span > < span class = "si" > %1.3g< / span > < span class = "s2" > " < / span > < span class = "o" > %< / span > < span class = "p" > (< / span > < span class = "n" > ave< / span > < span class = "p" > ,< / span > < span class = "n" > mi< / span > < span class = "p" > ,< / span > < span class = "n" > ma< / span > < span class = "p" > ))< / span >
2019-12-21 01:11:46 +00:00
< span class = "k" > return< / span > < span class = "n" > ave< / span >
2022-08-03 17:24:42 +00:00
2020-08-13 02:12:50 +00:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s2" > " Execution time for clr.predict" < / span > < span class = "p" > )< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > speed< / span > < span class = "p" > (< / span > < span class = "s2" > " clr.predict(X_test)" < / span > < span class = "p" > )< / span >
2020-08-13 02:12:50 +00:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s2" > " Execution time for ONNX Runtime" < / span > < span class = "p" > )< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > speed< / span > < span class = "p" > (< / span > < span class = "s2" > " sess.run([label_name], {input_name: X_test.astype(numpy.float32)})[0]" < / span > < span class = "p" > )< / span >
< / pre > < / div >
< / div >
< div class = "sphx-glr-script-out highlight-none notranslate" > < div class = "highlight" > < pre > < span > < / span > Execution time for clr.predict
2023-01-21 18:34:16 +00:00
Average 4.4e-05 min=4.25e-05 max=5.47e-05
2019-12-21 01:11:46 +00:00
Execution time for ONNX Runtime
2023-01-21 18:34:16 +00:00
Average 2.23e-05 min=2.16e-05 max=3.08e-05
2020-08-13 02:12:50 +00:00
2023-01-21 18:34:16 +00:00
2.2315799999432783e-05
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
< p > Let’
experiences: the model has to do one prediction at a time
as opposed to a batch of prediction.< / p >
2020-08-13 02:12:50 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "k" > def< / span > < span class = "nf" > loop< / span > < span class = "p" > (< / span > < span class = "n" > X_test< / span > < span class = "p" > ,< / span > < span class = "n" > fct< / span > < span class = "p" > ,< / span > < span class = "n" > n< / span > < span class = "o" > =< / span > < span class = "kc" > None< / span > < span class = "p" > ):< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > nrow< / span > < span class = "o" > =< / span > < span class = "n" > X_test< / span > < span class = "o" > .< / span > < span class = "n" > shape< / span > < span class = "p" > [< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span >
2020-08-13 02:12:50 +00:00
< span class = "k" > if< / span > < span class = "n" > n< / span > < span class = "ow" > is< / span > < span class = "kc" > None< / span > < span class = "p" > :< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > n< / span > < span class = "o" > =< / span > < span class = "n" > nrow< / span >
< span class = "k" > for< / span > < span class = "n" > i< / span > < span class = "ow" > in< / span > < span class = "nb" > range< / span > < span class = "p" > (< / span > < span class = "mi" > 0< / span > < span class = "p" > ,< / span > < span class = "n" > n< / span > < span class = "p" > ):< / span >
< span class = "n" > im< / span > < span class = "o" > =< / span > < span class = "n" > i< / span > < span class = "o" > %< / span > < span class = "n" > nrow< / span >
2022-08-03 17:24:42 +00:00
< span class = "n" > fct< / span > < span class = "p" > (< / span > < span class = "n" > X_test< / span > < span class = "p" > [< / span > < span class = "n" > im< / span > < span class = "p" > :< / span > < span class = "n" > im< / span > < span class = "o" > +< / span > < span class = "mi" > 1< / span > < span class = "p" > ])< / span >
2019-12-21 01:11:46 +00:00
2020-08-13 02:12:50 +00:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s2" > " Execution time for clr.predict" < / span > < span class = "p" > )< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > speed< / span > < span class = "p" > (< / span > < span class = "s2" > " loop(X_test, clr.predict, 100)" < / span > < span class = "p" > )< / span >
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "k" > def< / span > < span class = "nf" > sess_predict< / span > < span class = "p" > (< / span > < span class = "n" > x< / span > < span class = "p" > ):< / span >
< span class = "k" > return< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > run< / span > < span class = "p" > ([< / span > < span class = "n" > label_name< / span > < span class = "p" > ],< / span > < span class = "p" > {< / span > < span class = "n" > input_name< / span > < span class = "p" > :< / span > < span class = "n" > x< / span > < span class = "o" > .< / span > < span class = "n" > astype< / span > < span class = "p" > (< / span > < span class = "n" > numpy< / span > < span class = "o" > .< / span > < span class = "n" > float32< / span > < span class = "p" > )})[< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span >
2022-08-03 17:24:42 +00:00
2020-08-13 02:12:50 +00:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s2" > " Execution time for sess_predict" < / span > < span class = "p" > )< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > speed< / span > < span class = "p" > (< / span > < span class = "s2" > " loop(X_test, sess_predict, 100)" < / span > < span class = "p" > )< / span >
< / pre > < / div >
< / div >
< div class = "sphx-glr-script-out highlight-none notranslate" > < div class = "highlight" > < pre > < span > < / span > Execution time for clr.predict
2023-01-21 18:34:16 +00:00
Average 0.00404 min=0.00402 max=0.00413
2019-12-21 01:11:46 +00:00
Execution time for sess_predict
2023-01-21 18:34:16 +00:00
Average 0.00101 min=0.000995 max=0.00104
2020-08-13 02:12:50 +00:00
2023-01-21 18:34:16 +00:00
0.0010053269700003398
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
< p > Let’ < / p >
2020-08-13 02:12:50 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s2" > " Execution time for predict_proba" < / span > < span class = "p" > )< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > speed< / span > < span class = "p" > (< / span > < span class = "s2" > " loop(X_test, clr.predict_proba, 100)" < / span > < span class = "p" > )< / span >
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "k" > def< / span > < span class = "nf" > sess_predict_proba< / span > < span class = "p" > (< / span > < span class = "n" > x< / span > < span class = "p" > ):< / span >
< span class = "k" > return< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > run< / span > < span class = "p" > ([< / span > < span class = "n" > prob_name< / span > < span class = "p" > ],< / span > < span class = "p" > {< / span > < span class = "n" > input_name< / span > < span class = "p" > :< / span > < span class = "n" > x< / span > < span class = "o" > .< / span > < span class = "n" > astype< / span > < span class = "p" > (< / span > < span class = "n" > numpy< / span > < span class = "o" > .< / span > < span class = "n" > float32< / span > < span class = "p" > )})[< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span >
2022-08-03 17:24:42 +00:00
2020-08-13 02:12:50 +00:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s2" > " Execution time for sess_predict_proba" < / span > < span class = "p" > )< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > speed< / span > < span class = "p" > (< / span > < span class = "s2" > " loop(X_test, sess_predict_proba, 100)" < / span > < span class = "p" > )< / span >
< / pre > < / div >
< / div >
< div class = "sphx-glr-script-out highlight-none notranslate" > < div class = "highlight" > < pre > < span > < / span > Execution time for predict_proba
2023-01-21 18:34:16 +00:00
Average 0.00613 min=0.0061 max=0.00616
2019-12-21 01:11:46 +00:00
Execution time for sess_predict_proba
2023-01-21 18:34:16 +00:00
Average 0.00106 min=0.00106 max=0.0011
2020-08-13 02:12:50 +00:00
2023-01-21 18:34:16 +00:00
0.0010642362600015076
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
< p > This second comparison is better as
ONNX Runtime, in this experience,
computes the label and the probabilities
in every case.< / p >
2022-01-04 17:32:40 +00:00
< / section >
< section id = "benchmark-with-randomforest" >
2023-01-21 18:34:16 +00:00
< h2 > < a class = "toc-backref" href = "#id4" > Benchmark with RandomForest< / a > < a class = "headerlink" href = "#benchmark-with-randomforest" title = "Permalink to this heading" > ¶< / a > < / h2 >
2019-12-21 01:11:46 +00:00
< p > We first train and save a model in ONNX format.< / p >
2022-12-09 23:55:42 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "kn" > from< / span > < span class = "nn" > sklearn.ensemble< / span > < span class = "kn" > import< / span > < span class = "n" > RandomForestClassifier< / span >
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "n" > rf< / span > < span class = "o" > =< / span > < span class = "n" > RandomForestClassifier< / span > < span class = "p" > ()< / span >
< span class = "n" > rf< / span > < span class = "o" > .< / span > < span class = "n" > fit< / span > < span class = "p" > (< / span > < span class = "n" > X_train< / span > < span class = "p" > ,< / span > < span class = "n" > y_train< / span > < span class = "p" > )< / span >
2022-08-03 17:24:42 +00:00
< span class = "n" > initial_type< / span > < span class = "o" > =< / span > < span class = "p" > [(< / span > < span class = "s2" > " float_input" < / span > < span class = "p" > ,< / span > < span class = "n" > FloatTensorType< / span > < span class = "p" > ([< / span > < span class = "mi" > 1< / span > < span class = "p" > ,< / span > < span class = "mi" > 4< / span > < span class = "p" > ]))]< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > onx< / span > < span class = "o" > =< / span > < span class = "n" > convert_sklearn< / span > < span class = "p" > (< / span > < span class = "n" > rf< / span > < span class = "p" > ,< / span > < span class = "n" > initial_types< / span > < span class = "o" > =< / span > < span class = "n" > initial_type< / span > < span class = "p" > )< / span >
< span class = "k" > with< / span > < span class = "nb" > open< / span > < span class = "p" > (< / span > < span class = "s2" > " rf_iris.onnx" < / span > < span class = "p" > ,< / span > < span class = "s2" > " wb" < / span > < span class = "p" > )< / span > < span class = "k" > as< / span > < span class = "n" > f< / span > < span class = "p" > :< / span >
< span class = "n" > f< / span > < span class = "o" > .< / span > < span class = "n" > write< / span > < span class = "p" > (< / span > < span class = "n" > onx< / span > < span class = "o" > .< / span > < span class = "n" > SerializeToString< / span > < span class = "p" > ())< / span >
< / pre > < / div >
< / div >
< p > We compare.< / p >
2022-01-04 17:32:40 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "n" > sess< / span > < span class = "o" > =< / span > < span class = "n" > rt< / span > < span class = "o" > .< / span > < span class = "n" > InferenceSession< / span > < span class = "p" > (< / span > < span class = "s2" > " rf_iris.onnx" < / span > < span class = "p" > ,< / span > < span class = "n" > providers< / span > < span class = "o" > =< / span > < span class = "n" > rt< / span > < span class = "o" > .< / span > < span class = "n" > get_available_providers< / span > < span class = "p" > ())< / span >
2019-12-21 01:11:46 +00:00
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "k" > def< / span > < span class = "nf" > sess_predict_proba_rf< / span > < span class = "p" > (< / span > < span class = "n" > x< / span > < span class = "p" > ):< / span >
< span class = "k" > return< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > run< / span > < span class = "p" > ([< / span > < span class = "n" > prob_name< / span > < span class = "p" > ],< / span > < span class = "p" > {< / span > < span class = "n" > input_name< / span > < span class = "p" > :< / span > < span class = "n" > x< / span > < span class = "o" > .< / span > < span class = "n" > astype< / span > < span class = "p" > (< / span > < span class = "n" > numpy< / span > < span class = "o" > .< / span > < span class = "n" > float32< / span > < span class = "p" > )})[< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span >
2022-08-03 17:24:42 +00:00
2020-08-13 02:12:50 +00:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s2" > " Execution time for predict_proba" < / span > < span class = "p" > )< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > speed< / span > < span class = "p" > (< / span > < span class = "s2" > " loop(X_test, rf.predict_proba, 100)" < / span > < span class = "p" > )< / span >
2020-08-13 02:12:50 +00:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "s2" > " Execution time for sess_predict_proba" < / span > < span class = "p" > )< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > speed< / span > < span class = "p" > (< / span > < span class = "s2" > " loop(X_test, sess_predict_proba_rf, 100)" < / span > < span class = "p" > )< / span >
< / pre > < / div >
< / div >
< div class = "sphx-glr-script-out highlight-none notranslate" > < div class = "highlight" > < pre > < span > < / span > Execution time for predict_proba
2023-01-21 18:34:16 +00:00
Average 0.682 min=0.68 max=0.685
2019-12-21 01:11:46 +00:00
Execution time for sess_predict_proba
2023-01-21 18:34:16 +00:00
Average 0.00132 min=0.0013 max=0.00137
2020-08-13 02:12:50 +00:00
2023-01-21 18:34:16 +00:00
0.001315003579999825
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
< p > Let’ < / p >
2020-08-13 02:12:50 +00:00
< div class = "highlight-default notranslate" > < div class = "highlight" > < pre > < span > < / span > < span class = "n" > measures< / span > < span class = "o" > =< / span > < span class = "p" > []< / span >
2019-12-21 01:11:46 +00:00
< span class = "k" > for< / span > < span class = "n" > n_trees< / span > < span class = "ow" > in< / span > < span class = "nb" > range< / span > < span class = "p" > (< / span > < span class = "mi" > 5< / span > < span class = "p" > ,< / span > < span class = "mi" > 51< / span > < span class = "p" > ,< / span > < span class = "mi" > 5< / span > < span class = "p" > ):< / span >
2020-08-13 02:12:50 +00:00
< span class = "nb" > print< / span > < span class = "p" > (< / span > < span class = "n" > n_trees< / span > < span class = "p" > )< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > rf< / span > < span class = "o" > =< / span > < span class = "n" > RandomForestClassifier< / span > < span class = "p" > (< / span > < span class = "n" > n_estimators< / span > < span class = "o" > =< / span > < span class = "n" > n_trees< / span > < span class = "p" > )< / span >
< span class = "n" > rf< / span > < span class = "o" > .< / span > < span class = "n" > fit< / span > < span class = "p" > (< / span > < span class = "n" > X_train< / span > < span class = "p" > ,< / span > < span class = "n" > y_train< / span > < span class = "p" > )< / span >
2022-08-03 17:24:42 +00:00
< span class = "n" > initial_type< / span > < span class = "o" > =< / span > < span class = "p" > [(< / span > < span class = "s2" > " float_input" < / span > < span class = "p" > ,< / span > < span class = "n" > FloatTensorType< / span > < span class = "p" > ([< / span > < span class = "mi" > 1< / span > < span class = "p" > ,< / span > < span class = "mi" > 4< / span > < span class = "p" > ]))]< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > onx< / span > < span class = "o" > =< / span > < span class = "n" > convert_sklearn< / span > < span class = "p" > (< / span > < span class = "n" > rf< / span > < span class = "p" > ,< / span > < span class = "n" > initial_types< / span > < span class = "o" > =< / span > < span class = "n" > initial_type< / span > < span class = "p" > )< / span >
< span class = "k" > with< / span > < span class = "nb" > open< / span > < span class = "p" > (< / span > < span class = "s2" > " rf_iris_< / span > < span class = "si" > %d< / span > < span class = "s2" > .onnx" < / span > < span class = "o" > %< / span > < span class = "n" > n_trees< / span > < span class = "p" > ,< / span > < span class = "s2" > " wb" < / span > < span class = "p" > )< / span > < span class = "k" > as< / span > < span class = "n" > f< / span > < span class = "p" > :< / span >
< span class = "n" > f< / span > < span class = "o" > .< / span > < span class = "n" > write< / span > < span class = "p" > (< / span > < span class = "n" > onx< / span > < span class = "o" > .< / span > < span class = "n" > SerializeToString< / span > < span class = "p" > ())< / span >
2022-01-04 17:32:40 +00:00
< span class = "n" > sess< / span > < span class = "o" > =< / span > < span class = "n" > rt< / span > < span class = "o" > .< / span > < span class = "n" > InferenceSession< / span > < span class = "p" > (< / span > < span class = "s2" > " rf_iris_< / span > < span class = "si" > %d< / span > < span class = "s2" > .onnx" < / span > < span class = "o" > %< / span > < span class = "n" > n_trees< / span > < span class = "p" > ,< / span > < span class = "n" > providers< / span > < span class = "o" > =< / span > < span class = "n" > rt< / span > < span class = "o" > .< / span > < span class = "n" > get_available_providers< / span > < span class = "p" > ())< / span >
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "k" > def< / span > < span class = "nf" > sess_predict_proba_loop< / span > < span class = "p" > (< / span > < span class = "n" > x< / span > < span class = "p" > ):< / span >
< span class = "k" > return< / span > < span class = "n" > sess< / span > < span class = "o" > .< / span > < span class = "n" > run< / span > < span class = "p" > ([< / span > < span class = "n" > prob_name< / span > < span class = "p" > ],< / span > < span class = "p" > {< / span > < span class = "n" > input_name< / span > < span class = "p" > :< / span > < span class = "n" > x< / span > < span class = "o" > .< / span > < span class = "n" > astype< / span > < span class = "p" > (< / span > < span class = "n" > numpy< / span > < span class = "o" > .< / span > < span class = "n" > float32< / span > < span class = "p" > )})[< / span > < span class = "mi" > 0< / span > < span class = "p" > ]< / span >
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "n" > tsk< / span > < span class = "o" > =< / span > < span class = "n" > speed< / span > < span class = "p" > (< / span > < span class = "s2" > " loop(X_test, rf.predict_proba, 100)" < / span > < span class = "p" > ,< / span > < span class = "n" > number< / span > < span class = "o" > =< / span > < span class = "mi" > 5< / span > < span class = "p" > ,< / span > < span class = "n" > repeat< / span > < span class = "o" > =< / span > < span class = "mi" > 5< / span > < span class = "p" > )< / span >
< span class = "n" > trt< / span > < span class = "o" > =< / span > < span class = "n" > speed< / span > < span class = "p" > (< / span > < span class = "s2" > " loop(X_test, sess_predict_proba_loop, 100)" < / span > < span class = "p" > ,< / span > < span class = "n" > number< / span > < span class = "o" > =< / span > < span class = "mi" > 5< / span > < span class = "p" > ,< / span > < span class = "n" > repeat< / span > < span class = "o" > =< / span > < span class = "mi" > 5< / span > < span class = "p" > )< / span >
2022-08-03 17:24:42 +00:00
< span class = "n" > measures< / span > < span class = "o" > .< / span > < span class = "n" > append< / span > < span class = "p" > ({< / span > < span class = "s2" > " n_trees" < / span > < span class = "p" > :< / span > < span class = "n" > n_trees< / span > < span class = "p" > ,< / span > < span class = "s2" > " sklearn" < / span > < span class = "p" > :< / span > < span class = "n" > tsk< / span > < span class = "p" > ,< / span > < span class = "s2" > " rt" < / span > < span class = "p" > :< / span > < span class = "n" > trt< / span > < span class = "p" > })< / span >
2019-12-21 01:11:46 +00:00
2022-12-09 23:55:42 +00:00
< span class = "kn" > from< / span > < span class = "nn" > pandas< / span > < span class = "kn" > import< / span > < span class = "n" > DataFrame< / span >
2022-08-03 17:24:42 +00:00
2019-12-21 01:11:46 +00:00
< span class = "n" > df< / span > < span class = "o" > =< / span > < span class = "n" > DataFrame< / span > < span class = "p" > (< / span > < span class = "n" > measures< / span > < span class = "p" > )< / span >
2020-08-13 02:12:50 +00:00
< span class = "n" > ax< / span > < span class = "o" > =< / span > < span class = "n" > df< / span > < span class = "o" > .< / span > < span class = "n" > plot< / span > < span class = "p" > (< / span > < span class = "n" > x< / span > < span class = "o" > =< / span > < span class = "s2" > " n_trees" < / span > < span class = "p" > ,< / span > < span class = "n" > y< / span > < span class = "o" > =< / span > < span class = "s2" > " sklearn" < / span > < span class = "p" > ,< / span > < span class = "n" > label< / span > < span class = "o" > =< / span > < span class = "s2" > " scikit-learn" < / span > < span class = "p" > ,< / span > < span class = "n" > c< / span > < span class = "o" > =< / span > < span class = "s2" > " blue" < / span > < span class = "p" > ,< / span > < span class = "n" > logy< / span > < span class = "o" > =< / span > < span class = "kc" > True< / span > < span class = "p" > )< / span >
2022-08-03 17:24:42 +00:00
< span class = "n" > df< / span > < span class = "o" > .< / span > < span class = "n" > plot< / span > < span class = "p" > (< / span > < span class = "n" > x< / span > < span class = "o" > =< / span > < span class = "s2" > " n_trees" < / span > < span class = "p" > ,< / span > < span class = "n" > y< / span > < span class = "o" > =< / span > < span class = "s2" > " rt" < / span > < span class = "p" > ,< / span > < span class = "n" > label< / span > < span class = "o" > =< / span > < span class = "s2" > " onnxruntime" < / span > < span class = "p" > ,< / span > < span class = "n" > ax< / span > < span class = "o" > =< / span > < span class = "n" > ax< / span > < span class = "p" > ,< / span > < span class = "n" > c< / span > < span class = "o" > =< / span > < span class = "s2" > " green" < / span > < span class = "p" > ,< / span > < span class = "n" > logy< / span > < span class = "o" > =< / span > < span class = "kc" > True< / span > < span class = "p" > )< / span >
2019-12-21 01:11:46 +00:00
< span class = "n" > ax< / span > < span class = "o" > .< / span > < span class = "n" > set_xlabel< / span > < span class = "p" > (< / span > < span class = "s2" > " Number of trees" < / span > < span class = "p" > )< / span >
< span class = "n" > ax< / span > < span class = "o" > .< / span > < span class = "n" > set_ylabel< / span > < span class = "p" > (< / span > < span class = "s2" > " Prediction time (s)" < / span > < span class = "p" > )< / span >
< span class = "n" > ax< / span > < span class = "o" > .< / span > < span class = "n" > set_title< / span > < span class = "p" > (< / span > < span class = "s2" > " Speed comparison between scikit-learn and ONNX Runtime< / span > < span class = "se" > \n< / span > < span class = "s2" > For a random forest on Iris dataset" < / span > < span class = "p" > )< / span >
< span class = "n" > ax< / span > < span class = "o" > .< / span > < span class = "n" > legend< / span > < span class = "p" > ()< / span >
< / pre > < / div >
< / div >
2022-08-03 17:24:42 +00:00
< img src = "../images/sphx_glr_plot_train_convert_predict_001.png" srcset = "../images/sphx_glr_plot_train_convert_predict_001.png" alt = "Speed comparison between scikit-learn and ONNX Runtime For a random forest on Iris dataset" class = "sphx-glr-single-img" / > < div class = "sphx-glr-script-out highlight-none notranslate" > < div class = "highlight" > < pre > < span > < / span > 5
2023-01-21 18:34:16 +00:00
Average 0.0493 min=0.0492 max=0.0495
Average 0.00102 min=0.001 max=0.00105
2019-12-21 01:11:46 +00:00
10
2023-01-21 18:34:16 +00:00
Average 0.0827 min=0.0826 max=0.0828
Average 0.00103 min=0.00102 max=0.00105
2019-12-21 01:11:46 +00:00
15
2023-01-21 18:34:16 +00:00
Average 0.116 min=0.116 max=0.116
Average 0.00103 min=0.00102 max=0.00105
2019-12-21 01:11:46 +00:00
20
2023-01-21 18:34:16 +00:00
Average 0.149 min=0.149 max=0.149
Average 0.00106 min=0.00105 max=0.00108
2019-12-21 01:11:46 +00:00
25
2023-01-21 18:34:16 +00:00
Average 0.182 min=0.182 max=0.182
Average 0.00107 min=0.00106 max=0.00109
2019-12-21 01:11:46 +00:00
30
2023-01-21 18:34:16 +00:00
Average 0.215 min=0.215 max=0.215
Average 0.00109 min=0.00108 max=0.00112
2019-12-21 01:11:46 +00:00
35
2023-01-21 18:34:16 +00:00
Average 0.249 min=0.248 max=0.249
Average 0.00108 min=0.00107 max=0.00111
2019-12-21 01:11:46 +00:00
40
2023-01-21 18:34:16 +00:00
Average 0.282 min=0.282 max=0.283
Average 0.00111 min=0.0011 max=0.00113
2019-12-21 01:11:46 +00:00
45
2023-01-21 18:34:16 +00:00
Average 0.315 min=0.314 max=0.315
Average 0.00112 min=0.0011 max=0.00114
2019-12-21 01:11:46 +00:00
50
2023-01-21 18:34:16 +00:00
Average 0.348 min=0.348 max=0.348
Average 0.00114 min=0.00113 max=0.00116
2020-08-13 02:12:50 +00:00
2023-01-21 18:34:16 +00:00
< matplotlib.legend.Legend object at 0x7f94760bd870>
2019-12-21 01:11:46 +00:00
< / pre > < / div >
< / div >
2023-01-21 18:34:16 +00:00
< p class = "sphx-glr-timing" > < strong > Total running time of the script:< / strong > ( 3 minutes 10.085 seconds)< / p >
2022-08-03 17:24:42 +00:00
< div class = "sphx-glr-footer sphx-glr-footer-example docutils container" id = "sphx-glr-download-auto-examples-plot-train-convert-predict-py" >
2020-08-13 02:12:50 +00:00
< div class = "sphx-glr-download sphx-glr-download-python docutils container" >
2021-03-09 18:21:38 +00:00
< p > < a class = "reference download internal" download = "" href = "../downloads/c647c128e0cf2b3db04ce60b41ef1a14/plot_train_convert_predict.py" > < code class = "xref download docutils literal notranslate" > < span class = "pre" > Download< / span > < span class = "pre" > Python< / span > < span class = "pre" > source< / span > < span class = "pre" > code:< / span > < span class = "pre" > plot_train_convert_predict.py< / span > < / code > < / a > < / p >
2019-12-21 01:11:46 +00:00
< / div >
2020-08-13 02:12:50 +00:00
< div class = "sphx-glr-download sphx-glr-download-jupyter docutils container" >
2021-03-09 18:21:38 +00:00
< p > < a class = "reference download internal" download = "" href = "../downloads/1680115d3d937dfbb2d86adb705d9c5d/plot_train_convert_predict.ipynb" > < code class = "xref download docutils literal notranslate" > < span class = "pre" > Download< / span > < span class = "pre" > Jupyter< / span > < span class = "pre" > notebook:< / span > < span class = "pre" > plot_train_convert_predict.ipynb< / span > < / code > < / a > < / p >
2019-12-21 01:11:46 +00:00
< / div >
< / div >
2020-08-13 02:12:50 +00:00
< p class = "sphx-glr-signature" > < a class = "reference external" href = "https://sphinx-gallery.github.io" > Gallery generated by Sphinx-Gallery< / a > < / p >
2022-01-04 17:32:40 +00:00
< / section >
< / section >
2019-12-21 01:11:46 +00:00
< / div >
< / div >
< / div >
< div class = "sphinxsidebar" role = "navigation" aria-label = "main navigation" >
< div class = "sphinxsidebarwrapper" >
< p class = "logo" > < a href = "../index.html" >
2021-02-11 00:34:31 +00:00
< img class = "logo" src = "../static/ONNX_Runtime_icon.png" alt = "Logo" / >
2019-12-21 01:11:46 +00:00
< / a > < / p >
< h1 class = "logo" > < a href = "../index.html" > ONNX Runtime< / a > < / h1 >
< h3 > Navigation< / h3 >
2021-03-09 18:21:38 +00:00
< ul >
2019-12-21 01:11:46 +00:00
< li class = "toctree-l1" > < a class = "reference internal" href = "../tutorial.html" > Tutorial< / a > < / li >
2022-03-16 01:13:11 +00:00
< li class = "toctree-l1" > < a class = "reference internal" href = "../api_summary.html" > API< / a > < / li >
2021-03-09 18:21:38 +00:00
< li class = "toctree-l1" > < a class = "reference internal" href = "index.html" > Gallery of examples< / a > < / li >
2019-12-21 01:11:46 +00:00
< / ul >
< div class = "relations" >
< h3 > Related Topics< / h3 >
< ul >
< li > < a href = "../index.html" > Documentation overview< / a > < ul >
< li > < a href = "index.html" > Gallery of examples< / a > < ul >
< li > Previous: < a href = "plot_common_errors.html" title = "previous chapter" > Common errors with onnxruntime< / a > < / li >
< / ul > < / li >
< / ul > < / li >
< / ul >
< / div >
< div id = "searchbox" style = "display: none" role = "search" >
< h3 id = "searchlabel" > Quick search< / h3 >
< div class = "searchformwrapper" >
< form class = "search" action = "../search.html" method = "get" >
2022-01-04 17:32:40 +00:00
< input type = "text" name = "q" aria-labelledby = "searchlabel" autocomplete = "off" autocorrect = "off" autocapitalize = "off" spellcheck = "false" / >
2019-12-21 01:11:46 +00:00
< input type = "submit" value = "Go" / >
< / form >
< / div >
< / div >
2022-08-03 17:24:42 +00:00
< script > document . getElementById ( 'searchbox' ) . style . display = "block" < / script >
2020-04-06 22:52:48 +00:00
2019-12-21 01:11:46 +00:00
< / div >
< / div >
< div class = "clearer" > < / div >
< / div >
< div class = "footer" >
2023-01-21 18:34:16 +00:00
© 2018-2023, Microsoft.
2019-12-21 01:11:46 +00:00
|
2023-01-21 18:34:16 +00:00
Powered by < a href = "http://sphinx-doc.org/" > Sphinx 5.3.0< / a >
2023-01-20 00:19:31 +00:00
& < a href = "https://github.com/bitprophet/alabaster" > Alabaster 0.7.13< / a >
2019-12-21 01:11:46 +00:00
|
2021-02-11 00:34:31 +00:00
< a href = "../sources/auto_examples/plot_train_convert_predict.rst.txt"
2019-12-21 01:11:46 +00:00
rel="nofollow">Page source< / a >
< / div >
< / body >
< / html >
