From 0a2797a9c0cda417c1e4c6f04bf5f35603c1d37b Mon Sep 17 00:00:00 2001 From: Christopher Teubert Date: Wed, 7 Aug 2024 12:02:40 -0700 Subject: [PATCH] Multiple metrics --- src/progpy/data_models/lstm_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/progpy/data_models/lstm_model.py b/src/progpy/data_models/lstm_model.py index dfaedf7..0449ad4 100644 --- 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(