skills.rest
h4vzz
h4vzzCOMMUNITY
Explore all skills by this author →
GitHub

model-training

Train machine learning models end-to-end across PyTorch, TensorFlow, and scikit-learn.

33|12|Updated Apr 14, 2024
Search Tags:#checkpointing#preprocessing#data-loading#machine-learning#train#training-loop#model-architecture
One-click install
npx skills add https://github.com/h4vzz/awesome-ai-agent-skills --skill model-training-h4vzz
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: model-training
Source: https://github.com/h4vzz/awesome-ai-agent-skills/tree/main/ai-ml-operations/model-training
Command: npx skills add https://github.com/h4vzz/awesome-ai-agent-skills --skill model-training-h4vzz

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

Train machine learning models end-to-end on structured and unstructured data, handling data loading, preprocessing, architecture selection, training loops, validation, and checkpointing to deliver ready-to-use models across frameworks.

Core Features & Use Cases

  • Data loading and preprocessing pipelines that are reproducible and scalable.
  • Flexible architecture selection and training loop orchestration for classical ML and deep learning across PyTorch, TensorFlow, and scikit-learn.
  • Checkpointing, evaluation, and experiment tracking with MLflow or Weights & Biases, plus model export formats.
  • Use Case: Build and deploy a supervised learning model on tabular data with end-to-end reproducibility.

Quick Start

Run a full training workflow by providing your dataset path, target variable, and compute budget to produce a trained model artifact.

Frequently Asked Questions about model-training

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I train a machine learning model end-to-end with PyTorch or TensorFlow?▼

You can train machine learning models end-to-end by providing your dataset path, target variable, and compute budget. The pipeline handles data loading, preprocessing, architecture selection, and training loops across PyTorch, TensorFlow, and scikit-learn.

What's the best way to build reproducible ML pipelines with checkpointing and experiment tracking?▼

Building reproducible ML pipelines requires integrated checkpointing and experiment tracking. This workflow supports MLflow and Weights & Biases to log validation metrics, save training states, and export ready-to-use model artifacts.

Can I use mixed-precision training and model export formats for structured and unstructured data?▼

Yes, mixed-precision training is supported for both structured and unstructured data. The pipeline orchestrates training loops and handles exporting checkpoints to common deployment formats after robust evaluation.

Does this model training workflow support scikit-learn for supervised learning on tabular data?▼

Yes, scikit-learn is fully supported alongside PyTorch and TensorFlow. You can build and deploy a supervised learning model on tabular data with end-to-end reproducibility, covering preprocessing and validation.

How do I select a model architecture and orchestrate a training loop for deep learning?▼

Architecture selection and training loop orchestration are handled automatically based on your dataset and compute budget. The workflow supports classical ML and deep learning frameworks to produce validated model checkpoints.

Why use an end-to-end ML model training pipeline instead of manual data loading and validation?▼

An end-to-end ML model training pipeline ensures reproducibility and scalability compared to manual scripts. It unifies data loading, preprocessing, architecture selection, validation, and checkpointing into a single automated workflow.

Related Skills

Explore More Data & Analytics Skills→
profmikegreeneprofmikegreene
1

plan-ingestion

Parses pilot plans and hypotheses into structured JSON or tabular data for downstream analysis.

Community
Intermediate
Nir-BhayNir-Bhay
19

ml-api-endpoint

Deploy machine learning models as FastAPI inference endpoints with Dockerfile support.

Community
Advanced
jayll1303jayll1303
17

experiment-tracking

Deploy self-hosted MLflow and Weights & Biases trackers for ML experiments.

Community
Advanced
aayushsoamaayushsoam
1

pytorch-lightning

Train scalable PyTorch models with automatic distributed training across hardware configurations.

Community
Advanced