Abeshith's picture
Scripts Stages Added
cbb53b6

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