train_rf

Contents

train_rf#

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

Training/evaluation for RandomForest 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.

  • 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 RandomForestRegressor.

Returns:

y_hat_train, y_hat_test, model, y_train, y_test

Return type:

tuple

Contents