В настоящее время я пытаюсь предсказать движение частицы, используя предварительно созданный метод оценки тензорного потока (tf.estimator.DNNRegressor).
Я хочу сохранить изображение графика средних потерь, например, отображение одного тензорного щита, в папке каждой модели. 
Tensorboard достаточно хорош для мониторинга этого во время тренировки, но я хочу сохранить изображение для дальнейшего использования (например, визуальное сравнение различных подходов)
Есть ли простой способ сделать это?
Я мог бы сохранить результаты оценки в разное время и использовать matplotlib, но я не нашел ничего о том, как получить убыток от метода regressor.train.