Estefani Gaytan Nunez

upload

......@@ -37,10 +37,12 @@ __author__ = 'egaytan'
def savescreen(output, dic, path):
if output:
DF.from_dict(dic).to_csv(path+'.csv', sep = "\t", index = True)
def plotMax(yvalues, size):
def maxPoint(yvalues):
scorelist = list(yvalues)
maxscore = max(scorelist)
plt.plot(scorelist.index(maxscore), maxscore, markersize=size)
y = max(scorelist)
x = scorelist.index(y)
return(x,y)
#plt.plot(scorelist.index(maxscore), maxscore, markersize=size)
####################################################################################
# MAIN PROGRAM #
......@@ -94,7 +96,8 @@ if __name__ == '__main__':
plt.rcParams.update()
plt.grid()
plt.plot(scoresTable['precision'],'co--', label='precision', linewidth=1, markersize=4)
plotMax(scoresTable['precision'], 8)
x,y = maxPoint(scoresTable['precision'])
plt.plot(x, y, markersize=10)
plt.plot(scoresTable['f1-score'], 'bo--', label='F1', linewidth=1, markersize=4)
plt.plot(scoresTable['recall'], 'mo--', label='recall' , linewidth=1, markersize=4)
plt.plot(scoresTable['CV'], 'ro--', label='CV' , linewidth=1, markersize=4)
......