Rubyでguesslangのニューラルネットワークモデルを読み込みたい

Python製のプログラミング言語検出ツールでguesslangというのがある。

これをRubyでやりたくて、そのモデルを読み込みたい。

TensorFlowのSavedModel形式とやらで配布されているので、それを読めればいいわけだ。

tensorflow-rubyでは、SavedModel形式対応は計画には入っていてその為のブランチもあるけど、まだできてはいない様子（のまま五年が過ぎている……）。

Stack Overflowのpython - Convert Tensorflow Saved Model format to Keras .h5 model formatというページではSavedModelをKeras API（という言い方でいいのか？）でも読み込めそうで、tensorflow-rubyはKerasとやらは実装してそうなのでいけるかも知れない。試す。

Guesslangではこうなっている：

code:model.py

def load(saved_model_dir: str) -> AutoTrackable:

"""Load a Tensorflow saved model"""

tf.saved_model.load(saved_model_dir)

Kerasで読み込むにはこうらしい：

code:python

new_model = tf.keras.models.load_model('Path of the saved model along with the model name')

tensorflow-rubyのREADMによると

This library follows the TensorFlow 2.0 Python API.

とのことなので何となくで試してみる。

……が、Tf::Keras::Modelsには特に何も定義されていなかった。

TensorflowのPython実装見てみたら、Kerasのload_modelは結局Tensorflowのload_modelを呼んでいる。

SavedModelをHDF5に変換して読み込むならいけるか？と思ったが、結局それでも読み込むAPIが今は無い。

Kerasのload_modelはh5pyとやらに処理を委譲している。

h5pyは（当然ながら）TensowflowのC APIを呼び出しているわけではない（TensorflowにHDF5読み込み機能があるわけではない）のでtensorflow-rubyを育てれば読めるという話ではない。

HDF5を読み込むgemはあるが、そのRuby APIとtensorflow-rubyのAPIの間を埋める必要がある。

てかHdf5 gem、作りかけなんじゃ・・・

と思ったら今時はTensorFlowのモデルをONNXというフォーマットに変換して使えということみたい： TensorFlow Object Detection in Ruby

PythonとTensorflowのバージョンに指定があるので注意