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

Training Plotting code: Add minor gridlines.

This commit is contained in:
Brandon Amos 2016-03-18 14:10:53 -04:00
parent 02884dde26
commit b30d6d9461
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
training/plot-loss.py
View File

@ -59,7 +59,8 @@ def plot(workDirs):
plt.ylim(ymin=0) plt.ylim(ymin=0)
# plt.xlim(xmin=1) # plt.xlim(xmin=1)
plt.grid(b=True, which='major', color='k', linestyle='-') plt.grid(b=True, which='major', color='k', linestyle='-')
plt.grid(b=True, which='minor', color='k', linestyle='--', alpha=0.2) plt.grid(b=True, which='minor', color='k', linestyle='-', alpha=0.2)
plt.minorticks_on()
# ax.set_yscale('log') # ax.set_yscale('log')
d = os.path.join(plotDir, "train-loss.pdf") d = os.path.join(plotDir, "train-loss.pdf")
fig.savefig(d) fig.savefig(d)
@ -71,6 +72,9 @@ def plot(workDirs):
plt.xlabel("Epoch") plt.xlabel("Epoch")
plt.ylabel("LFW Accuracy") plt.ylabel("LFW Accuracy")
plt.ylim(ymin=0, ymax=1) plt.ylim(ymin=0, ymax=1)
plt.grid(b=True, which='major', color='k', linestyle='-')
plt.grid(b=True, which='minor', color='k', linestyle='-', alpha=0.2)
plt.minorticks_on()
# plt.xlim(xmin=1) # plt.xlim(xmin=1)
# ax.set_yscale('log') # ax.set_yscale('log')
d = os.path.join(plotDir, "lfw-accuracy.pdf") d = os.path.join(plotDir, "lfw-accuracy.pdf")