Spaces:
Running
Running
Utility Scripts
Simple CLI scripts for common operations.
Available Scripts
1. Train Pipeline
Run the complete training pipeline (all 7 stages):
python scripts/train.py
2. Evaluate Model
Evaluate the trained model on test data:
# Use default model path
python scripts/evaluate.py
# Or specify custom model path
python scripts/evaluate.py --model-path artifacts/model_trainer/model
3. Start API Server
Start the FastAPI server:
# Default (localhost:8000)
python scripts/serve.py
# Custom host/port
python scripts/serve.py --host 0.0.0.0 --port 8080
# Development mode with auto-reload
python scripts/serve.py --reload
4. Initialize Airflow
Set up Airflow database and create admin user:
python scripts/init_db.py
Then start Airflow services:
airflow scheduler # Terminal 1
airflow webserver # Terminal 2
Quick Examples
# Full workflow
python scripts/train.py # Train model
python scripts/evaluate.py # Evaluate model
python scripts/serve.py --reload # Start API server
# Airflow setup
python scripts/init_db.py # Initialize
airflow webserver # Start UI