Update app.py

app.py

@@ -247,10 +247,10 @@ def visualize_ROC(set_threshold,set_input):
   plt.ylim(0, 1)
   #xi = (c1 - c2) / (m2 - m1)
   #yi = m1 * xi + c1
-  #plt.axvline(x=set_threshold, color='gray', linestyle='--')
-  #plt.axhline(y=f1_score_cur, color='gray', linestyle='--')
-  #plt.scatter(set_threshold, f1_score_cur, color='red', s=300)
-  #plt.scatter(set_threshold, g_mean_cur, color='red', s=300)
+  plt.axvline(x=set_threshold, color='gray', linestyle='--')
+  plt.axhline(y=f1_score_cur, color='gray', linestyle='--')
+  plt.scatter(set_threshold, f1_score_cur, color='red', s=300)
+  plt.scatter(set_threshold, g_mean_cur, color='red', s=300)
 
   ax.set_facecolor("white")
 
@@ -263,8 +263,8 @@ def visualize_ROC(set_threshold,set_input):
   plt.xlabel('Threshold cut-off')
   plt.ylabel('F1-score & G-mean')
   plt.legend(loc='upper left')
-  #plt.text(set_threshold, f1_score_cur, 'F1-score:%s' % (round(f1_score_cur,2)))
-  #plt.text(set_threshold, g_mean_cur, 'G-mean:%s' % (round(g_mean_cur,2)))
+  plt.text(set_threshold, f1_score_cur, 'F1-score:%s' % (round(f1_score_cur,2)))
+  plt.text(set_threshold, g_mean_cur, 'G-mean:%s' % (round(g_mean_cur,2)))
   plt.title("Threshold tuning curves (F1-score & G-mean)", fontsize=20)
   plt.tight_layout()