未验证 提交 cd9374fb 编写于 作者: Z zhangchao 提交者: GitHub

Update plotlog.py

上级 41b75add
......@@ -70,7 +70,10 @@ def sample(metric, sample_rate):
return metric_sample
def plot_metric(metric, batch_id, graph_title, line_style='b-',
def plot_metric(metric,
batch_id,
graph_title,
line_style='b-',
line_label='y',
line_num=1):
plt.figure()
......@@ -78,7 +81,7 @@ def plot_metric(metric, batch_id, graph_title, line_style='b-',
if line_num == 1:
plt.plot(batch_id, metric, line_style, line_label)
else:
for i in line_num:
for i in range(line_num):
plt.plot(batch_id, metric[i], line_style[i], line_label[i])
plt.xlabel('batch')
plt.ylabel(graph_title)
......@@ -99,7 +102,11 @@ def main():
accuracy_sample = sample(accuracy, args.sample_rate)
plot_metric(loss_sample, batch_sample, 'loss', line_label='loss')
plot_metric(accuracy_sample, batch_sample, 'accuracy', line_style='g-', line_label='accuracy')
plot_metric(accuracy_sample,
batch_sample,
'accuracy',
line_style='g-',
line_label='accuracy')
if __name__ == '__main__':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册