こしあん
2018-07-18

Kerasでモデルのsummaryをテキストとして保存する方法

Kerasで「plot_modelを使えばモデルの可視化ができるが、GraphViz入れないといけなかったり、セットアップが面倒くさい!model.summary()のテキストをファイル保存で十分だ!」という場合に使えるテクニックです。

summary()のprint_fn引数を使う

summary()はコンソールに出力するための関数で、出力の文字列を返したりすることはできません(返り値はNoneです)。ただし、print_fn引数を使うと出力先を書き換えることができ、これによりファイル出力ができます。

Kerasのsummary()関数を見ると次のように書かれています。なお、model.summary()はkeras.utils.print_summaryへのショートカットです。

print_fn: 使うためのプリント関数.サマリの各行で呼ばれます.サマリの文字列をキャプチャするためにカスタム関数を指定することもできます.

https://keras.io/ja/utils/#print_summary

ファイルに出力する場合はこうします。

with open("output.txt", "w") as fp:
    model.summary(print_fn=lambda x: fp.write(x + "\r\n"))

print_fnにラムダ式をコールバックの形で指定します。行ごとに呼ばれるので、末尾に改行コードをつけないと改行がないテキストファイルが出力されます。改行コードは環境にあわせて切り替えてください(\nにするなど)。

summaryの文字列を取得するには

同じように、summary()の文字列の値を取得するには、io.StringIOを使うとできます。

from io import StringIO

with StringIO() as buf:
    # StringIOに書き込む
    model.summary(print_fn=lambda x: buf.write(x + "\n"))
    # StringIOから取得
    text = buf.getvalue()

print(text)

結果的にはmodel.summary()と何ら変わりませんが、一度変数に記録した値を表示しているのが異なるところです。StringIOの使い方については詳しくは公式ドキュメントを参照してください。ちなみに最新バージョンではモジュールがStringIOから、io.StringIOに変わっているのでご注意ください。

https://docs.python.jp/2/library/stringio.html
https://docs.python.org/ja/3.7/library/io.html

関連

Related Posts

tensorflow.kerasでKeras方式のhdf5で重みを保存する方法... 従来のKerasで係数を保存すると「hdf5」形式で保存されたのですが、TPU環境などでTensorFlowのKerasAPIを使うと、TensorFlow形式のチェックポイントまるごと保存で互換性の面で困ったことがおきます。従来のKerasのhdf5形式で保存する方法を紹介します。 サンプル...
Google ColabのTPU環境でmodel.fitのhistoryが消える現象... Google ColabのTPU環境でmodel.fitしたときに、通常の環境で得られるhistory(誤差や精度のログ)が消えていることがあります。その対応法を示します。 原因はTPU用のモデルに変換したから まず結論からいうとこの現象はCPU/GPU環境では再発しません。TPU環境特有の現...
Kerasに組み込まれているVGG16/19の実装 VGG-16のsummary _________________________________________________________________ Layer (type) Output Shape Param # ==...
Kerasに組み込まれているMobileNetの実装 MobileNetのsummary _________________________________________________________________ Layer (type) Output Shape Param # ...
PCA Color Augmentationを拡張してTensorFlow/Keras向けに実装した... PCA Color AugmentationはAlexNetの論文に示された画像向けのData Augmentationですが、画像用だけではなく、テンソルの固有値分解をすることで構造化データに対しても使えるようにしてみました。これの解説と効果を書きます。 リポジトリ こちら https://...
Tags:

Add a Comment

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です