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)")
|