Training: plot-loss: Show last few values.

krishachetan89 · Oct 11, 2015 · 245a833 · 245a833
1 parent 3e5d1a6
commit 245a833
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/training/plot-loss.py b/training/plot-loss.py
@@ -46,7 +46,10 @@ def plot(workDirs):
     trainDf = pd.concat(trainDfs, ignore_index=True)
     testDf = pd.concat(testDfs, ignore_index=True)
 
-    print(testDf)
+    print("train, test:")
+    print("\n".join(["{:0.2e}, {:0.2e}".format(x,y) for (x,y) in
+                     zip(trainDf['avg triplet loss (train set)'].values[-5:],
+                         testDf['avg triplet loss (test set)'].values[-5:])]))
 
     fig, ax = plt.subplots(1,1)
     trainDf.index += 1
@@ -69,5 +72,4 @@ def plot(workDirs):
     parser = argparse.ArgumentParser()
     parser.add_argument('workDirs', type=int, nargs='+')
     args = parser.parse_args()
-    print(args.workDirs)
     plot(args.workDirs)