Updating the benchmark figures to say glum instead of quantcore-glm (#…

…451) * Updating the benchmark figures to say glum instead of quantcore-glm * Fix black issues.
Quantco · Oct 8, 2021 · 5224f98 · 5224f98
1 parent aa22946
commit 5224f98
Show file tree

Hide file tree

Showing 19 changed files with 4 additions and 1 deletion.
diff --git a/docs/_static/headline_benchmark.pdf b/docs/_static/headline_benchmark.pdf
diff --git a/docs/_static/headline_benchmark.png b/docs/_static/headline_benchmark.png
diff --git a/docs/_static/intermediate-housing-l2.pdf b/docs/_static/intermediate-housing-l2.pdf
diff --git a/docs/_static/intermediate-housing-l2.png b/docs/_static/intermediate-housing-l2.png
diff --git a/docs/_static/intermediate-housing-lasso.pdf b/docs/_static/intermediate-housing-lasso.pdf
diff --git a/docs/_static/intermediate-housing-lasso.png b/docs/_static/intermediate-housing-lasso.png
diff --git a/docs/_static/intermediate-insurance-l2.pdf b/docs/_static/intermediate-insurance-l2.pdf
diff --git a/docs/_static/intermediate-insurance-l2.png b/docs/_static/intermediate-insurance-l2.png
diff --git a/docs/_static/intermediate-insurance-lasso.pdf b/docs/_static/intermediate-insurance-lasso.pdf
diff --git a/docs/_static/intermediate-insurance-lasso.png b/docs/_static/intermediate-insurance-lasso.png
diff --git a/docs/_static/narrow-insurance-l2.pdf b/docs/_static/narrow-insurance-l2.pdf
diff --git a/docs/_static/narrow-insurance-l2.png b/docs/_static/narrow-insurance-l2.png
diff --git a/docs/_static/narrow-insurance-lasso.pdf b/docs/_static/narrow-insurance-lasso.pdf
diff --git a/docs/_static/narrow-insurance-lasso.png b/docs/_static/narrow-insurance-lasso.png
diff --git a/docs/_static/wide-insurance-l2.pdf b/docs/_static/wide-insurance-l2.pdf
diff --git a/docs/_static/wide-insurance-l2.png b/docs/_static/wide-insurance-l2.png
diff --git a/docs/_static/wide-insurance-lasso.pdf b/docs/_static/wide-insurance-lasso.pdf
diff --git a/docs/_static/wide-insurance-lasso.png b/docs/_static/wide-insurance-lasso.png
diff --git a/docs/benchmarks/benchmark_figure.py b/docs/benchmarks/benchmark_figure.py
@@ -6,7 +6,7 @@
 #       extension: .py
 #       format_name: percent
 #       format_version: '1.3'
-#       jupytext_version: 1.11.4
+#       jupytext_version: 1.13.0
 #   kernelspec:
 #     display_name: Python 3 (ipykernel)
 #     language: python
@@ -101,6 +101,9 @@
         plot_df = plot_df.pivot(columns="library_name")
         plot_df.columns = plot_df.columns.get_level_values(1)
         plot_df.index = [x.title() for x in plot_df.index]
+        plot_df = plot_df[["h2o", "glum", "r-glmnet"]].rename(
+            columns={"r-glmnet": "glmnet"}
+        )
 
         title = prob_name.title() + "-" + ("Lasso" if reg == "lasso" else "Ridge")
         plot_df.plot.bar(