Question

有人可以告诉我Tensorboard是否支持从命令行导出CSV文件？我问这个的原因是因为我有很多日志目录，我希望有一个脚本文件来自动化该过程。感谢。

Answer 1

API支持以编程方式读取文件。这是一个提取标签数据并将其保存到.csv的示例，格式与tensorboard

生成的格式相似

import argparse
import numpy as np
import tensorflow as tf

def save_tag_to_csv(fn, tag='test_metric', output_fn=None):
    if output_fn is None:
        output_fn = '{}.csv'.format(tag.replace('/', '_'))
    print("Will save to {}".format(output_fn))

    sess = tf.InteractiveSession()

    wall_step_values = []
    with sess.as_default():
        for e in tf.train.summary_iterator(fn):
            for v in e.summary.value:
                if v.tag == tag:
                    wall_step_values.append((e.wall_time, e.step, v.simple_value))
    np.savetxt(output_fn, wall_step_values, delimiter=',', fmt='%10.5f')

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('fn')
    parser.add_argument('--tag', default='test_metric')
    args = parser.parse_args()
    save_tag_to_csv(args.fn, tag=args.tag)

Tensorboard：从命令行导出CSV文件

1 个答案: