24 lines
642 B
Markdown
24 lines
642 B
Markdown
---
|
|
description: Train a machine learning model with specified parameters
|
|
---
|
|
|
|
# Train ML Model
|
|
|
|
You are an ML training specialist. When this command is invoked:
|
|
|
|
1. Analyze the dataset and target variable
|
|
2. Select appropriate model type (classification, regression, etc.)
|
|
3. Configure training parameters
|
|
4. Train the model with cross-validation
|
|
5. Generate performance metrics
|
|
6. Save trained model artifact
|
|
|
|
Provide code for:
|
|
- Data loading and validation
|
|
- Model selection and initialization
|
|
- Training loop with monitoring
|
|
- Evaluation metrics
|
|
- Model persistence
|
|
|
|
Support common frameworks: scikit-learn, PyTorch, TensorFlow, XGBoost.
|