こしあん
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形式で保存する方法を紹介します。 サンプル...
BrestCancerデータセットをCNNで分類する 構造化データを畳み込みニューラルネットワーク(CNN)で分析することを考えます。BrestCancerデータセットはScikit-learnに用意されている、乳がんが良性か悪性かの2種類を分類する典型的な構造化データです。サンプル数569、データの次元30の典型的な構造化データです。 なぜ畳み込...
Kerasに組み込まれているResNet50の実装 ResNet50のsummary __________________________________________________________________________________________________ Layer (type) O...
Google ColabのTPU環境でmodel.fitのhistoryが消える現象... Google ColabのTPU環境でmodel.fitしたときに、通常の環境で得られるhistory(誤差や精度のログ)が消えていることがあります。その対応法を示します。 原因はTPU用のモデルに変換したから まず結論からいうとこの現象はCPU/GPU環境では再発しません。TPU環境特有の現...
Kerasに組み込まれているNASNet(Large)の実装... NASNet Largeのsummary __________________________________________________________________________________________________ Layer (type) ...
Tags:

Add a Comment

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