sklearn-model-trainer

Solid

Scikit-learn model training skill with cross-validation, hyperparameter tuning, pipeline construction, and model serialization. Enables automated ML model development using scikit-learn's comprehensive toolkit.

AI & Automation 814 stars 53 forks Updated today MIT

Install

View on GitHub

Quality Score: 95/100

Stars 20%
97
Recency 20%
100
Frontmatter 20%
70
Documentation 15%
100
Issue Health 10%
50
License 10%
100
Description 5%
100

Skill Content

# Scikit-learn Model Trainer Train machine learning models using scikit-learn with cross-validation, hyperparameter tuning, and pipeline construction. ## Overview This skill provides comprehensive capabilities for training machine learning models using scikit-learn. It supports the full model development workflow from data preprocessing through model training, evaluation, and serialization. ## Capabilities ### Model Training - Train classification models (LogisticRegression, RandomForest, SVM, etc.) - Train regression models (LinearRegression, GradientBoosting, etc.) - Train clustering models (KMeans, DBSCAN, etc.) - Support for ensemble methods (VotingClassifier, Stacking, etc.) ### Cross-Validation - K-fold cross-validation - Stratified K-fold for imbalanced datasets - Time series split for temporal data - Leave-one-out and leave-p-out validation - Custom cross-validation strategies ### Hyperparameter Tuning - GridSearchCV for exhaustive search - RandomizedSearchCV for random sampling - Halving search strategies for efficiency - Custom scoring functions - Multi-metric evaluation ### Pipeline Construction - Feature preprocessing pipelines - Column transformers for heterogeneous data - Feature selection integration - Composite pipelines with caching ### Model Serialization - Save models with joblib (recommended) - Pickle serialization - ONNX export for interoperability - Model versioning support ## Prerequisites ### Installation ```bash pip install scikit-learn>=1....

Details

Author
a5c-ai
Repository
a5c-ai/babysitter
Created
4 months ago
Last Updated
today
Language
JavaScript
License
MIT

Related Skills