fiddle with plots
Browse files
app.py
CHANGED
|
@@ -148,9 +148,15 @@ def get_plot(model_name, plot_eager, generate_type):
|
|
| 148 |
df = df[df["framework"] != "TF (Eager Execition)"]
|
| 149 |
|
| 150 |
g = sns.catplot(
|
| 151 |
-
data=df,
|
| 152 |
-
|
| 153 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 154 |
)
|
| 155 |
g.despine(left=True)
|
| 156 |
g.set_axis_labels("GPU", "Generation time (ms)")
|
|
|
|
| 148 |
df = df[df["framework"] != "TF (Eager Execition)"]
|
| 149 |
|
| 150 |
g = sns.catplot(
|
| 151 |
+
data=df,
|
| 152 |
+
kind="bar",
|
| 153 |
+
x="variable",
|
| 154 |
+
y="value",
|
| 155 |
+
hue="framework",
|
| 156 |
+
ci="sd",
|
| 157 |
+
palette={"PyTorch": "blue", "TF (Eager Execition)": "orange", "TF (XLA)": "red"},
|
| 158 |
+
alpha=.6,
|
| 159 |
+
height=6
|
| 160 |
)
|
| 161 |
g.despine(left=True)
|
| 162 |
g.set_axis_labels("GPU", "Generation time (ms)")
|