train_sisso_ga

Contents

train_sisso_ga#

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

Training/evaluation for SISSO model with GA-based feature selection.

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.

  • 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 GA feature selection and SISSO inputs.

Return type:

tuple[ndarray, ndarray, SISSORegressor, ndarray, ndarray]

Contents