TensorFlowでLossを詳しく解析する方法 ②カスタムメトリックを記録する

前回の続きです。 前回: bluepost69.hatenablog.com おさらい:簡単にメトリックを取得する方法 前回カスタムメトリックを実装する方法をいくつか示しました。単に学習中のカスタムメトリックを取得するだけなら、train_stepのreturnに取得したいメトリックの値を入れておけばよいです。 def train_step(self, data): # …