gmmcleod commited on
Commit
664c476
·
verified ·
1 Parent(s): b642cf8
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -191,7 +191,7 @@ def compute_stats_sector(sectors_model, sector_gt):
191
  else:
192
  TN += 1
193
  NUM_SECTORS = 8 - ignored
194
- return [round(TP / NUM_SECTORS,2), round(FP / NUM_SECTORS,2), round(FN / NUM_SECTORS,2), round(TN / NUM_SECTORS,2)]
195
 
196
  #Compare the model output with ground truth
197
  #return TP, FP, FN, TN
@@ -298,7 +298,7 @@ def evaluate_model(username, groupname, file):
298
  return leaderboard_data.values.tolist()
299
  print("calculating new entry")
300
 
301
- new_entry = pd.DataFrame([[username, groupname, execution_time, accuracy, TP, FP, FN, TN]],
302
  columns=HEADERS)
303
  print("updating new entry")
304
  leaderboard_data = update_results_dataset(new_entry)
 
191
  else:
192
  TN += 1
193
  NUM_SECTORS = 8 - ignored
194
+ return [TP / NUM_SECTORS, FP / NUM_SECTORS, FN / NUM_SECTORS, TN / NUM_SECTORS]
195
 
196
  #Compare the model output with ground truth
197
  #return TP, FP, FN, TN
 
298
  return leaderboard_data.values.tolist()
299
  print("calculating new entry")
300
 
301
+ new_entry = pd.DataFrame([[username, groupname, round(execution_time,2), round(accuracy,2), round(TP,2), round(FP,2), round(FN,2), round(TN,2)]],
302
  columns=HEADERS)
303
  print("updating new entry")
304
  leaderboard_data = update_results_dataset(new_entry)