#!/bin/bash
TJA_DIR=${1:-"sample/training"}
MODEL_PATH=${2:-"sample/model"}
DATASET_DIR=${3:-"output/dataset"}

mkdir -p "$MODEL_PATH" "$DATASET_DIR" "sample/factorize"

echo "--- 분석 및 학습 시작 ---"
for f in $(find "$TJA_DIR" -name "*.tja"); do bun run scripts/factorize.ts "$f" > /dev/null; done
bun run scripts/prepare_dataset.ts "$DATASET_DIR"
bun run scripts/train.ts "$MODEL_PATH" "$DATASET_DIR"