Multiple metrics

nasa · Aug 7, 2024 · 0a2797a · 0a2797a
1 parent 0a8f7c1
commit 0a2797a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/progpy/data_models/lstm_model.py b/src/progpy/data_models/lstm_model.py
@@ -572,7 +572,7 @@ def from_data(cls, inputs, outputs, event_states=None, t_met=None, **kwargs):
  output_data.append(t_all)
 
  model = keras.Model(inputs, outputs)
- model.compile(optimizer="rmsprop", loss="mse", metrics=["mae"])
+ model.compile(optimizer="rmsprop", loss="mse", metrics=["mae"]*len(outputs))
 
  # Train model
  history = model.fit(