def train_single_frame_simple_model(): X_train, X_valid, Y_train, Y_valid = Datasets.make_single_frame_data( data_folder_path) model = simple_categorical() callbacks = [EarlyStopping(monitor='val_loss', patience=3)] #ModelCheckpoint(filepath='best_model.h5', monitor='val_loss', save_best_only=True)] model.fit(X_train, Y_train, epochs=3, batch_size=64, validation_data=(X_valid, Y_valid)) timestamp = datetime.datetime.now().strftime('%Y%m%d%H%M%S') model.save_weights('saved_models/single_frame_simple_categorical_' + timestamp + '.h5', overwrite=True)