소스 검색

fixing typo on x label

Beto 1 년 전
부모
커밋
7e7c8d2514
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      examples/plot_metrics.py

+ 2 - 2
examples/plot_metrics.py

@@ -47,11 +47,11 @@ def plot_metrics(file_path):
     directory = os.path.dirname(file_path)
     filename_prefix = os.path.basename(file_path).split('.')[0]
 
-    plot_metric(data, 'loss', 'Step', 'Loss', 'Loss', ['b', 'r'])
+    plot_metric(data, 'loss', 'Epoch', 'Loss', 'Loss', ['b', 'r'])
     plt.savefig(os.path.join(directory, f"{filename_prefix}_train_and_validation_loss.png"))
     plt.close()
 
-    plot_metric(data, 'perplexity', 'Step', 'Perplexity', 'Perplexity', ['g', 'm'])
+    plot_metric(data, 'perplexity', 'Epoch', 'Perplexity', 'Perplexity', ['g', 'm'])
     plt.savefig(os.path.join(directory, f"{filename_prefix}_train_and_validation_perplexity.png"))
     plt.close()