Skip to content
Snippets Groups Projects

Resolve "release v1.4.0"

Merged Ghost User requested to merge release_v1.4.0 into master
2 files
+ 43
61
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -82,7 +82,7 @@ class AbstractModelClass(ABC):
self.__custom_objects = value
@property
def compile_options(self) -> Callable:
def compile_options(self) -> Dict:
"""
The compile options property allows the user to use all keras.compile() arguments. They can ether be passed as
dictionary (1), as attribute, without setting compile_options (2) or as mixture (partly defined as instance
@@ -116,7 +116,7 @@ class AbstractModelClass(ABC):
def set_compile_options(self):
self.optimizer = keras.optimizers.SGD()
self.loss = keras.losses.mean_squared_error
self.compile_options = {"optimizer" = keras.optimizers.Adam(), "metrics": ["mse", "mae"]}
self.compile_options = {"optimizer": keras.optimizers.Adam(), "metrics": ["mse", "mae"]}
Note:
* As long as the attribute and the dict value have exactly the same values, the setter method will not raise
Loading