franceth commited on
Commit
7fb86f9
·
verified ·
1 Parent(s): c08d1fa

Fix value in radar plot

Browse files
Files changed (1) hide show
  1. app.py +2 -15
app.py CHANGED
@@ -1407,22 +1407,9 @@ with gr.Blocks(theme='shivi/calm_seafoam', css_paths='style.css', js=js_func) as
1407
  fig = go.Figure()
1408
  for model in sorted(selected_models, key=lambda m: avg_metrics[avg_metrics['model'] == m]['avg_metric'].mean(), reverse=True):
1409
  model_data = avg_metrics[avg_metrics['model'] == model]
1410
- # values = [
1411
- # model_data[model_data['test_category'] == cat]['avg_metric'].values[0]
1412
- # if cat in model_data['test_category'].values else 0
1413
- # for cat in categories
1414
- # ]
1415
  values = [
1416
- 0.4 if model in ["GPT-3.5", "Llama-8B", "DS-Llama3 70B"] and cat == "MANY-TO-MANY" else
1417
- 1.0 if model == "Llama-8B" and cat == "DISTINCT" else
1418
- 0.76 if model == "DS-Llama3 70B" and cat == "SELECT" else
1419
- 1.0 if model == "GPT-3.5" and cat == "Project" else
1420
- 0.89 if model == "Llama-8B" and cat == "Project" else
1421
- 0.87 if model == "GPT-3.5" and cat in model_data['test_category'].values else
1422
- 0.83 if model == "DS-Llama3 70B" and cat in model_data['test_category'].values else
1423
- 0.74 if model == "Llama-8B" and cat in model_data['test_category'].values else
1424
- (model_data[model_data['test_category'] == cat]['avg_metric'].values[0]
1425
- if cat in model_data['test_category'].values else 0)
1426
  for cat in categories
1427
  ]
1428
  fig.add_trace(go.Scatterpolar(
 
1407
  fig = go.Figure()
1408
  for model in sorted(selected_models, key=lambda m: avg_metrics[avg_metrics['model'] == m]['avg_metric'].mean(), reverse=True):
1409
  model_data = avg_metrics[avg_metrics['model'] == model]
 
 
 
 
 
1410
  values = [
1411
+ model_data[model_data['test_category'] == cat]['avg_metric'].values[0]
1412
+ if cat in model_data['test_category'].values else 0
 
 
 
 
 
 
 
 
1413
  for cat in categories
1414
  ]
1415
  fig.add_trace(go.Scatterpolar(