diff --git a/src/progpy/data_models/lstm_model.py b/src/progpy/data_models/lstm_model.py index 58ec924e..853338f7 100644 --- a/src/progpy/data_models/lstm_model.py +++ b/src/progpy/data_models/lstm_model.py @@ -537,9 +537,7 @@ def from_data(cls, inputs, outputs, event_states=None, t_met=None, **kwargs): from tensorflow import keras # Build model - callbacks = [ - keras.callbacks.ModelCheckpoint("best_model.keras", save_best_only=True) - ] + callbacks = [ ] if params['early_stop']: callbacks.append(keras.callbacks.EarlyStopping(**params['early_stop.cfg'])) @@ -592,8 +590,6 @@ def from_data(cls, inputs, outputs, event_states=None, t_met=None, **kwargs): workers=params['workers'], use_multiprocessing=(params['workers'] > 1)) - model = keras.models.load_model("best_model.keras") - # Split model into separate models n_state_layers = params['layers'] + 1 + (params['dropout'] > 0) + (params['normalize']) output_layer_input = keras.layers.Input(model.layers[n_state_layers-1].output.shape[1:])