sasha HF staff commited on
Commit
2f8fc9d
β€’
1 Parent(s): b7b78a8
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -8,6 +8,7 @@ def get_plots(task):
8
  task_df= pd.read_csv('data/energy/'+task)
9
  params_df = pd.read_csv('data/params/'+task)
10
  all_df = pd.merge(task_df, params_df, on='Link'))
 
11
  all_df['Total GPU Energy (Wh)'] = all_df['total_gpu_energy']*1000
12
  all_df = task_df.sort_values(by=['Total GPU Energy (Wh)'])
13
  all_df['energy_star'] = pd.cut(all_df['Total GPU Energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])
 
8
  task_df= pd.read_csv('data/energy/'+task)
9
  params_df = pd.read_csv('data/params/'+task)
10
  all_df = pd.merge(task_df, params_df, on='Link'))
11
+ print(all_df.head())
12
  all_df['Total GPU Energy (Wh)'] = all_df['total_gpu_energy']*1000
13
  all_df = task_df.sort_values(by=['Total GPU Energy (Wh)'])
14
  all_df['energy_star'] = pd.cut(all_df['Total GPU Energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])