train_modnet

Contents

train_modnet#

mlproject.training.fold_trainer.train_modnet(X_train, X_test, y_train, y_test, target_name, n_jobs, save_model, fold_ind, **kwargs)[source]#

Training/evaluation for MODNet model.

Parameters:

  • X_train (pd.DataFrame) – Training features.

  • X_test (pd.DataFrame) – Testing features.

  • y_train (pd.DataFrame) – Training target.

  • y_test (pd.DataFrame) – Testing target.

  • target_name (str) – Name of the target variable.

  • n_jobs (int) – Number of parallel jobs.

  • save_model (bool) – Whether to save the trained model.

  • fold_ind (int) – Fold index for cross-validation.

  • **kwargs – Additional keyword arguments for MODNet’s genetic algorithm hyperparameter optimization.

Return type:

tuple[ndarray, ndarray, EnsembleMODNetModel, ndarray, ndarray]

Contents