sklearn-model-trainer - Scikit-learn Model Training
Train scikit-learn ML models with cross-validation, hyperparameter tuning, and pipelines
Tags
Updated: 2026-05-09Capabilities
Typical Inputs
Typical Outputs
What this skill does
- Train classification models
- Train regression models
- Train clustering models
- Perform cross-validation
- Tune hyperparameters
- Build preprocessing pipelines
- Serialize trained models
Inputs
- Training dataset
- Target column name
- Model type
- Hyperparameters
- Scoring metric
Outputs
- Trained model file
- Cross-validation scores
- Evaluation metrics
Requirements
- scikit-learn >= 1.0.0
- joblib
- pandas
- numpy