05141045

models_processing/model_tf/tf_lstm_zone.py

@@ -55,7 +55,7 @@ class TSHandler(object):
 
         model = Model(nwp_input, [zone, output])
         adam = optimizers.Adam(learning_rate=0.001, beta_1=0.9, beta_2=0.999, epsilon=1e-7, amsgrad=True)
-        model.compile(loss={"zone": loss, "cdq_output": loss}, loss_weights={"zone": 0.1, "cdq_output": 0.9}, optimizer=adam)
+        model.compile(loss={"zone": loss, "cdq_output": loss}, loss_weights={"zone": 0.5, "cdq_output": 0.5}, optimizer=adam)
 
         return model