compare_models
Compare multiple model types using cross-validation and get a ranked table to select the best model before hyperparameter tuning.
Instructions
Compare multiple model types using cross-validation. Returns a ranked table. Quick way to find the best model type before fine-tuning hyperparameters. Default models: linear_regression, random_forest, gradient_boosting (or classifiers if target is categorical). Example: compare_models(target_column="Revenue", model_types=["linear_regression","random_forest","gradient_boosting"])
Input Schema
| Name | Required | Description | Default |
|---|---|---|---|
| df_name | No | ||
| n_folds | No | ||
| model_types | No | ||
| target_column | Yes |
Output Schema
| Name | Required | Description | Default |
|---|---|---|---|
| result | Yes |