scikit-learnやlightgbmなどのONNX形式モデルがブラウザで動作しないとき

やりたいこと

Pythonで訓練したモデルをONNX形式でブラウザ上（JavaScript）で推論させたい。

使用ライブラリ

• Python
  • scikit-learn/lightgbm
  • onnxmltools
• JavaScript
  • onnxruntime-web

問題

JSのサンプルコード通りに書いてもsession.run()部分で412398８のようなエラーが出て動かない。

解決法

PythonでモデルをONNX形式に変換する際、zipmapオプションを無効にしないとブラウザでは動かないらしい。
例えば、LightGBMモデルだとこんな感じ。

import onnxmltools
onnx_model = onnxmltools.convert_lightgbm(model, initial_types=initial_types, zipmap=False)

参考