Agent skill · research science · aj-geddes
ML Model Training
Build and train machine learning models using scikit-learn, PyTorch, and TensorFlow for classification, regression, and clustering tasks
Why this skill is useful
Provides specific implementation details and domain knowledge for training ML models using popular libraries like scikit-learn, PyTorch, and TensorFlow.
What it needs
Requires matplotlib, numpy, pandas, scikit-learn, tensorflow, torch installed locally. About 3k tokens when loaded. Last updated 2026-03-04. 307 stars on the source repository.
What this skill does
ML Model Training Training machine learning models involves selecting appropriate algorithms, preparing data, and optimizing model parameters to achieve strong predictive performance. Training Phases Data Preparation: Cleaning, encoding, normalization Feature Engineering: Creating meaningful features Model Selection: Choosing appropriate algorithms Hyperparameter Tuning: Optimizing model settings Validation: Cross-validation and evaluation metrics Deployment: Preparing models for production Common Algorithms Regression: Linear, Ridge, Lasso, Random Forest Classification: Logistic, SVM, Random Forest, Gradient Boosting Clustering: K-Means, DBSCAN, Hierarchical Neural Networks: MLPs, CNNs, RNNs, Transformers Python Implementation Training Best Practices Data Split: 70/15/15 for train/validation/test Scaling: Normalize features before training Cross-validation: Use K-fold for robust evaluation Early Stopping: Prevent overfitting Class Balancing: Handle imbalanced datasets Key Metrics Accuracy: Overall correctness Precision: Positive prediction accuracy Recall: True positive detection rate F1 Score: Harmonic mean of precision/recall ROC-AUC: Threshold-independent metric Deliverables Trained model checkpoint Performance metrics on test set Feature importance analysis Learning curves Hyperparameter configuration Model evaluation report
How to use it
Reference it in AdaL, Claude Code, Cursor or any coding agent — nothing to install:
@skills aj-geddes/ml-model-training