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

Fix flake8 warnings.

This commit is contained in:
Brandon Amos 2016-01-06 17:19:37 -05:00
parent 7923c2785c
commit 4821cb6ee2
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
training/plot-loss.py
View File

@ -23,7 +23,6 @@ plt.style.use('bmh')
import pandas as pd
import os
import sys
scriptDir = os.path.dirname(os.path.realpath(__file__))
plotDir = os.path.join(scriptDir, 'plots')
@ -56,8 +55,7 @@ def plot(workDirs):
# 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, alpha=0.6)
plt.legend(['Train loss, semi-hard triplets'])
# 'Test loss, random triplets'])
plt.legend(['Train loss, semi-hard triplets']) # 'Test loss, random triplets'])
plt.xlabel("Epoch")
plt.ylabel("Loss")
# plt.ylim(ymin=0)