ML model training — end-to-end training pipeline with scikit-learn, PyTorch and TensorFlow
Full training script comparing Logistic Regression, Random Forest, Gradient Boosting, a PyTorch MLP and a Keras network on a synthetic dataset, with metrics and comparison charts
- What
- Full training script comparing Logistic Regression, Random Forest, Gradient Boosting, a PyTorch MLP and a Keras network on a synthetic dataset, with metrics and comparison charts
- Cost
- Free
- Needs
- Python with scikit-learn, numpy, pandas, matplotlib; torch and tensorflow for the deep-learning parts — a real dataset to train on for real work
- Install
- Copy the installer prompt below into your Muse — your agent does the rest.
Curated by Skill Harbor — @aj-geddes's ML model training skill: a complete, runnable training pipeline in one file. Generates a synthetic binary-classification dataset, then trains and compares five models head to head — Logistic Regression, Random Forest, Gradient Boosting (scikit-learn), a PyTorch MLP with dropout, and a Keras Sequential network — with standard scaling, train/test split, accuracy/precision/recall/F1/ROC-AUC metrics, training-loss curves, and comparison charts. Wraps up with training best practices (splits, scaling, cross-validation, early stopping, class balancing) and a metrics cheat sheet. Honest caveats: it trains on a synthetic dataset to demonstrate the pipeline — swap in your own data for real work; deep-learning parts need torch/tensorflow installed. MIT licensed. Skill Harbor never reviews the code, review it yourself before use. Discovered via skills.sh.
Version:
Install
Prerequisites: Python with scikit-learn, numpy, pandas, matplotlib; torch and tensorflow for the deep-learning parts — a real dataset to train on for real work Install "ML model training — end-to-end training pipeline with scikit-learn, PyTorch and TensorFlow" for me. It gives my agent @aj-geddes's complete training pipeline: synthetic dataset generation, standard scaling and train/test split, head-to-head training of Logistic Regression, Random Forest, Gradient Boosting, a PyTorch MLP and a Keras network with accuracy/precision/recall/F1/ROC-AUC metrics, loss curves and comparison charts, plus training best practices and a metrics cheat sheet. MIT-licensed. Repository: https://github.com/aj-geddes/useful-ai-prompts/blob/main/skills/ml-model-training/SKILL.md 1. Fetch the SKILL.md file (and any helper files) from the repository path into a temporary folder and summarize what it does in one or two sentences. 2. Safety check: review the SKILL.md and scripts for anything suspicious (unexpected network calls, shell commands, credential harvesting). This repo should contain zero secrets in code. Verify that holds here; STOP on any red flag and tell me. 3. Install it as a skill: copy SKILL.md and its helper files into the agent's skills directory, in a folder named "ml-model-training". 4. Verify with no network calls: frontmatter valid, files in place. 5. Report what was installed, where, and what I still need to do myself (e.g. pip install scikit-learn torch tensorflow matplotlib and tell the agent whether to run the synthetic demo or adapt the pipeline to my own dataset; nothing else — it's a walkthrough with runnable code). GitHub is optional: if I have a GitHub account or the gh CLI, you may use it; otherwise public access is fine. Never require it unless it's in the prerequisites above. Rules: don't touch anything outside the temp folder and the install target. If anything looks off, stop and ask me.
Questions
How do I install a build?
Every product page includes a copy-paste install prompt. Paste it into your Muse and it sets the build up for you — no manual configuration.
Where does my money go?
Straight to the seller. Skill Harbor never processes payments: checkout happens on the seller’s own page, usually Stripe.
What does the ✓ next to a creator’s name mean?
It means we confirmed the identity of the person behind the listing. It says nothing about the code itself — always check a build before installing it.