From c7d7146e541b20e889a51efb4de5894957e3a68f Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Wed, 23 Sep 2020 09:28:55 +0200
Subject: [PATCH] update modelclass example
---
Examples_from_manuscript.ipynb | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/Examples_from_manuscript.ipynb b/Examples_from_manuscript.ipynb
index 2c5cca61..5c89cd9a 100644
--- a/Examples_from_manuscript.ipynb
+++ b/Examples_from_manuscript.ipynb
@@ -118,10 +118,10 @@
" A customised model with a 1x1 Conv, and 2 Dense layers (16, \n",
" output shape). Dropout is used after Conv layer.\n",
" \"\"\"\n",
- " def __init__(self, shape_inputs: list, shape_outputs: list):\n",
+ " def __init__(self, input_shape: list, output_shape: list):\n",
" \n",
" # set attributes shape_inputs and shape_outputs\n",
- " super().__init__(shape_inputs[0], shape_outputs[0])\n",
+ " super().__init__(input_shape[0], output_shape[0])\n",
"\n",
" # apply to model\n",
" self.set_model()\n",
--
GitLab