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 `_, `tensorflow `_, -`sklearn-onnx `_, +`keras-onnx `_, `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())