Update Evaluator.py

Making classId work also as a string
NnaYelsel · Apr 16, 2019 · aca5279 · aca5279
1 parent 110242c
commit aca5279
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/Evaluator.py b/lib/Evaluator.py
@@ -226,7 +226,7 @@ def PlotPrecisionRecallCurve(self,
                 # ap_str = "{0:.4f}%".format(average_precision * 100)
                 plt.title('Precision x Recall curve \nClass: %s, AP: %s' % (str(classId), ap_str))
             else:
-                plt.title('Precision x Recall curve \nClass: %d' % classId)
+                plt.title('Precision x Recall curve \nClass: %s' % str(classId))
             plt.legend(shadow=True)
             plt.grid()
             ############################################################