OrgLion-ML
/
openface
mirror of https://github.com/cmusatyalab/openface.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Training: Plot: Set alpha on the test loss since it overlaps the train loss.

This commit is contained in:
Brandon Amos 2015-10-03 22:47:27 -04:00
parent 4509fac2e3
commit f51d02fb51
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
training/plot-loss.py
View File

@ -52,7 +52,7 @@ def plot(workDirs):
trainDf.index += 1
testDf.index += 1
trainDf['avg triplet loss (train set)'].plot(legend='True', ax=ax)
testDf['avg triplet loss (test set)'].plot(legend='True', ax=ax)
testDf['avg triplet loss (test set)'].plot(legend='True', ax=ax, alpha=0.6)
plt.legend(['Train loss, semi-hard triplets',
'Test loss, random triplets'])
plt.xlabel("Epoch")