From d33dbb23b23c1d1f5f9ab76091ad412ed1d055eb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Xavier=20Dupr=C3=A9?= <xadupre@users.noreply.github.com>
Date: Fri, 7 Jun 2019 12:03:46 +0200
Subject: [PATCH] replace onnxmltools by keras-onnx in one example (#1151)

---
 docs/python/examples/plot_dl_keras.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/python/examples/plot_dl_keras.py b/docs/python/examples/plot_dl_keras.py
index 3f41d15931..5cea5d1f15 100644
--- a/docs/python/examples/plot_dl_keras.py
+++ b/docs/python/examples/plot_dl_keras.py
@@ -14,7 +14,7 @@ of a pretrained deep learning model obtained from
 with *onnxruntime*. The conversion requires
 `keras <https://keras.io/>`_,
 `tensorflow <https://www.tensorflow.org/>`_,
-`sklearn-onnx <https://github.com/onnx/sklearn-onnx/>`_,
+`keras-onnx <https://github.com/onnx/keras-onnx/>`_,
 `onnxmltools <https://pypi.org/project/onnxmltools/>`_
 but then only *onnxruntime* is required
 to compute the predictions.
@@ -24,7 +24,7 @@ if not os.path.exists('dense121.onnx'):
     from keras.applications.densenet import DenseNet121
     model = DenseNet121(include_top=True, weights='imagenet')
 
-    from onnxmltools import convert_keras
+    from keras2onnx import convert_keras
     onx = convert_keras(model, 'dense121.onnx')    
     with open("dense121.onnx", "wb") as f:
         f.write(onx.SerializeToString())