sasha HF staff commited on
Commit
15710e6
·
1 Parent(s): 308e2ba

calculating energy star before

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,11 +34,11 @@ def get_all_plots():
34
  params_df= params_df.rename(columns={"Link": "model"})
35
  tasks_df = pd.merge(task_df, params_df, on='model')
36
  tasks_df= tasks_df[['model', 'parameters', 'total_gpu_energy']]
 
 
37
  all_df = pd.concat([all_df, tasks_df])
38
- all_df['Total GPU Energy (Wh)'] = all_df['total_gpu_energy']*1000
39
  all_df = all_df.sort_values(by=['Total GPU Energy (Wh)'])
40
  all_df['parameters'] = all_df['parameters'].apply(format_params)
41
- all_df['energy_star'] = pd.cut(all_df['Total GPU Energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])
42
  fig = px.scatter(all_df, x="model", y='Total GPU Energy (Wh)', custom_data=['parameters'], height= 500, width= 800, color = 'energy_star', color_discrete_map={"⭐": 'red', "⭐⭐": "yellow", "⭐⭐⭐": "green"})
43
  fig.update_traces(
44
  hovertemplate="<br>".join([
 
34
  params_df= params_df.rename(columns={"Link": "model"})
35
  tasks_df = pd.merge(task_df, params_df, on='model')
36
  tasks_df= tasks_df[['model', 'parameters', 'total_gpu_energy']]
37
+ tasks_df['Total GPU Energy (Wh)'] = tasks_df['total_gpu_energy']*1000
38
+ tasks_df['energy_star'] = pd.cut(tasks_df['Total GPU Energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])
39
  all_df = pd.concat([all_df, tasks_df])
 
40
  all_df = all_df.sort_values(by=['Total GPU Energy (Wh)'])
41
  all_df['parameters'] = all_df['parameters'].apply(format_params)
 
42
  fig = px.scatter(all_df, x="model", y='Total GPU Energy (Wh)', custom_data=['parameters'], height= 500, width= 800, color = 'energy_star', color_discrete_map={"⭐": 'red', "⭐⭐": "yellow", "⭐⭐⭐": "green"})
43
  fig.update_traces(
44
  hovertemplate="<br>".join([