def call(self, trial: TrialSplit, epoch_number: int, total_epochs: int, input_train: DataContainer,
pred_train: DataContainer, input_val: DataContainer, pred_val: DataContainer,
is_finished_and_fitted: bool):
trial.save_model()
return False
def call(self, trial_split: TrialSplit, epoch_number: int, total_epochs: int, input_train: DataContainer,
pred_train: DataContainer, input_val: DataContainer, pred_val: DataContainer, context: ExecutionContext,
is_finished_and_fitted: bool):
trial_split.save_model(self.label)
return False