sasha HF staff commited on
Commit
7675047
·
1 Parent(s): 0620faf
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ def get_plots(task):
7
  #TO DO : hover text with energy efficiency number, parameters
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
 
7
  #TO DO : hover text with energy efficiency number, parameters
8
  task_df= pd.read_csv('data/energy/'+task)
9
  params_df = pd.read_csv('data/params/'+task)
10
+ task_df= task_df.rename(columns={"model": "Link"})
11
  all_df = pd.merge(task_df, params_df, on='Link')
12
  print(all_df.head())
13
  all_df['Total GPU Energy (Wh)'] = all_df['total_gpu_energy']*1000