Spaces:
Running
Running
calculating energy star before
Browse files
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([
|