update for keras 3.0

aimat-lab · Nov 13, 2023 · 6597501 · 6597501
1 parent ef6284a
commit 6597501
Show file tree

Hide file tree

Showing 4 changed files with 326 additions and 40 deletions.
diff --git a/README.md b/README.md
@@ -136,7 +136,7 @@ class MyMessageNN(MessagePassingBase):
     def __init__(self, units, **kwargs):
         super(MyMessageNN, self).__init__(**kwargs)
         self.dense = ks.layers.Dense(units)
-        self.add = ks.layers.Add(axis=-1)
+        self.add = ks.layers.Add()
 
     def message_function(self, inputs, **kwargs):
         n_in, n_out, edges = inputs

diff --git a/changelog.md b/changelog.md
@@ -1,6 +1,10 @@
 v4.0.0
 
-* 
+* Reworked training scripts to have a single ``train_graph.py`` script. Command line arguments are now optional and just used for verification, but `category` to select a model/hyperparameter combination from hyper file.
+Since the hyperparameter file already contains all necessary information.
+* Train test indices can now also be set and loaded from the dataset directly.
+* Scaler behaviour has changed with regard to `transform_dataset`. Key names of properties to transform has been moved to the constructor!
+
 
 
 v3.1.0