yasirs avatar yasirs committed c20fe05

legends in plot

Comments (0)

Files changed (1)


 	nr = netroc.NetReader(trainfile,testfile)
 	if labelList!=None:
 		labels = labelList
+	else:
+		labels = []
 	for scorefile in scorelist:
 		scores = nr.makeScores(scorefile)
 		d = yard.BinaryClassifierData(zip(scores,nr.getLabels()))
 		if (method.upper()=='ROC'):
 			result = yard.PrecisionRecallCurve(d)
 		pylab.plot(map(lambda x:x[0],result.points),map(lambda x:x[1],result.points))
-	pylab.legend()
+		if labelList==None:
+			labels.append(scorefile)
+	pylab.legend(labels)
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.