Makefile

UV ?= uv

.PHONY: setup validate-data dedupe-data split-data train eval export compile

setup:
	$(UV) sync

validate-data:
	$(UV) run python -m scripts.validate_labels --input data/raw/synthetic.jsonl

dedupe-data:
	$(UV) run python -m scripts.dedupe_examples --input data/raw/synthetic.jsonl --output data/raw/synthetic.deduped.jsonl

split-data:
	$(UV) run python -m scripts.split_dataset --input data/raw/synthetic.deduped.jsonl --output-dir data/synthetic --seed 13

# train:
# 	$(UV) run python -m scripts.train --train-file data/synthetic/train.jsonl --validation-file data/synthetic/validation.jsonl --output-dir artifacts/tiny-router --encoder-name microsoft/deberta-v3-small --batch-size 32 --epochs 20 --max-length 128

train:
	$(UV) run python -m scripts.train --train-file data/synthetic/train.jsonl --validation-file data/synthetic/validation.jsonl --output-dir artifacts/tiny-router --encoder-name microsoft/MiniLM-L12-H384-uncased --batch-size 32 --epochs 20 --max-length 128

# eval:
# 	$(UV) run python -m scripts.eval --model-dir artifacts/tiny-router --data-file data/synthetic/test.jsonl --output-file artifacts/tiny-router/eval.json --run-ablations

eval:
	$(UV) run python -m scripts.eval --model-dir artifacts/tiny-router --data-file data/synthetic/test.jsonl --output-file artifacts/tiny-router/eval.json --run-ablations

test:
	$(UV) run python -m scripts.predict --model-dir artifacts/tiny-router --input-json '{"current_text":"Actually next Monday","interaction":{"previous_text":"Set a reminder for Friday","previous_action":"created_reminder","previous_outcome":"success","recency_seconds":45}}' --pretty

# export:
# 	$(UV) run python -m scripts.export_onnx --model-dir artifacts/tiny-router --output-dir artifacts/tiny-router/onnx --quantize

export:
	$(UV) run python -m scripts.export_onnx --model-dir artifacts/tiny-router --output-dir artifacts/tiny-router/onnx --quantize

compile:
	$(UV) run python -m compileall tiny_router scripts