Skip to content
Snippets Groups Projects
Commit c9b79258 authored by v.gramlich1's avatar v.gramlich1
Browse files

minor_tail_loss=MSE

parent f9ce4697
No related branches found
No related tags found
1 merge request!302Draft: Resolve "Class-based Oversampling technique"
Pipeline #75861 passed
Stage: test
Stage: docs
Stage: pages
Stage: deploy
Hide whitespace changes
Inline Side-by-side
mlair/model_modules/model_class.py
+ 1
− 1
View file @ c9b79258
...@@ -458,7 +458,7 @@ class IntelliO3_ts_architecture(AbstractModelClass): ...@@ -458,7 +458,7 @@ class IntelliO3_ts_architecture(AbstractModelClass):
def set_compile_options(self): def set_compile_options(self):
self.compile_options = {"optimizer": keras.optimizers.adam(lr=self.initial_lr, amsgrad=True), self.compile_options = {"optimizer": keras.optimizers.adam(lr=self.initial_lr, amsgrad=True),
"loss": [l_p_loss(4), keras.losses.mean_squared_error], "loss": [keras.losses.mean_squared_error, keras.losses.mean_squared_error],
"metrics": ['mse'], "metrics": ['mse'],
"loss_weights": [.01, .99] "loss_weights": [.01, .99]
} }
\ No newline at end of file
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment