Makefile

.PHONY: help install data tce_train rlr_train evaluate test clean

help:
	@echo "AAMC Project Makefile"
	@echo "====================="
	@echo ""
	@echo "Available targets:"
	@echo "  install       - Install dependencies"
	@echo "  data          - Generate synthetic TCE training data"
	@echo "  tce_train     - Train Task Complexity Estimator"
	@echo "  rlr_train     - Train Reinforcement Learning Router"
	@echo "  evaluate      - Evaluate all routing strategies"
	@echo "  test          - Run unit tests"
	@echo "  clean         - Clean generated files"
	@echo ""
	@echo "Quick start:"
	@echo "  make install && make data && make tce_train"

install:
	pip install -r requirements.txt

data:
	python data/generate_tce_dataset.py --n_train 10000 --n_val 2000 --n_test 2000

data_small:
	python data/generate_tce_dataset.py --n_train 1000 --n_val 200 --n_test 200

tce_train:
	python tce/train_tce.py --config configs/tce.yaml --data_dir data

tce_eval:
	python tce/eval_tce.py --checkpoint checkpoints/tce/best_model.pt --data_path data/dtce_test.csv

rlr_train:
	python rler/train_rlr.py --config configs/rlr.yaml --sim_config configs/sim.yaml --tce_checkpoint checkpoints/tce/best_model.pt

rlr_train_quick:
	python rler/train_rlr.py --config configs/rlr.yaml --sim_config configs/sim.yaml --tce_checkpoint checkpoints/tce/best_model.pt --quick_test

evaluate:
	python scripts/evaluate_all.py --sim_config configs/sim.yaml --tce_checkpoint checkpoints/tce/best_model.pt --rlr_checkpoint checkpoints/rlr/final_model.pt --n_episodes 10

test:
	pytest tests/ -v --cov=. --cov-report=html

clean:
	rm -rf checkpoints/*
	rm -rf logs/*
	rm -rf experiments_results/*
	rm -rf __pycache__ */__pycache__ */*/__pycache__
	rm -rf .pytest_cache
	rm -rf htmlcov
	rm -rf *.egg-info