From 7b28f1ffddf03767ce59bf90e8ca7bfd75a91d72 Mon Sep 17 00:00:00 2001
From: Pat Alt <55311242+pat-alt@users.noreply.github.com>
Date: Tue, 8 Aug 2023 14:35:34 +0200
Subject: [PATCH] only robust NN left
---
notebooks/mnist.qmd | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/notebooks/mnist.qmd b/notebooks/mnist.qmd
index b5b68d58..c8a468ee 100644
--- a/notebooks/mnist.qmd
+++ b/notebooks/mnist.qmd
@@ -368,6 +368,7 @@ function _plot_eccco_mnist(
use_class_loss::Bool = true,
model_dict=model_dict,
wide::Bool = false,
+ img_height::Int = img_height,
)
# Setup:
@@ -500,12 +501,8 @@ add_retrain = true
# Deep Ensemble:
mlp_large_ens = EnsembleModel(model=mlp, n=50)
-# LeNet-5 Ensemble:
-lenet_ens = EnsembleModel(model=lenet, n=5)
-
add_models = Dict(
"LeNet-5" => lenet,
- "LeNet-5 Ensemble" => lenet_ens,
"Large Ensemble (n=50)" => mlp_large_ens,
)
@@ -524,7 +521,6 @@ _plt_order = [
"MLP Ensemble",
"Large Ensemble (n=50)",
"LeNet-5",
- "LeNet-5 Ensemble",
"JEM",
"JEM Ensemble",
]
@@ -537,6 +533,10 @@ display(plt_additional_models)
savefig(plt_additional_models, joinpath(output_images_path, "mnist_eccco_additional.png"))
```
+```{julia}
+
+```
+
### All digits
```{julia}
--
GitLab