TensorFlow.js で、モデルを保存する。ブラウザ版 #javascript #機械学習 #tensorflow

index:

概要:

前のTensorFlow.js 関係で、
学習済のモデルを、保存する例となります。

環境

TensorFlow.js
ブラウザ

参考

https://www.tensorflow.org/js/guide/save_load

保存の処理

Local Storage に、保存する場合ですが、他にもindexDBなど。あるようです

await model.save('localstorage://my-model');

例です

学習後、モデル保存し、
モデル読み込み、評価後、結果出力、

gist.github.com

node.js版

・環境: ubuntu16

npm install @tensorflow/tfjs-node

・保存
https://github.com/kuc-arc-f/tfjs_pred_2/blob/master/t6-save.js

   await model.save('file://test-model');

・ロード
https://github.com/kuc-arc-f/tfjs_pred_2/blob/master/t6-read.js

   const load_model = await tf.loadLayersModel('file://test-model/model.json');

参考のページ

knaka0209.hatenablog.com

knaka0209.hatenablog.com