From b0d484dbabf7189fa8c9cc653d46b43d82a40dcc Mon Sep 17 00:00:00 2001 From: Daniel Dolezal Date: Sat, 4 Apr 2026 00:00:33 +0200 Subject: [PATCH] add dataset and models folder into git ignore file and change train-ai to get the input data from the dataset --- .gitignore | 5 ++++- justfile | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index ab9f877..5fec197 100644 --- a/.gitignore +++ b/.gitignore @@ -10,4 +10,7 @@ data/ .tools/ dbschema/migrations/ -*.jsonl \ No newline at end of file + +*.jsonl +dataset/ +models/ diff --git a/justfile b/justfile index bf493c0..25ec178 100644 --- a/justfile +++ b/justfile @@ -171,7 +171,7 @@ curate-dataset input="good_moves-*.jsonl" output="data/dataset/best_moves.jsonl" analyze-dataset input="good_moves-*.jsonl" output="": if [ -n "{{output}}" ]; then python -m server.DatasetStats --input "{{input}}" --output "{{output}}"; else python -m server.DatasetStats --input "{{input}}"; fi -train-ai input="data/dataset/best_moves.jsonl" rl_input="data/dataset/rl_bootstrap.jsonl" output="models/battlesnake_softmax_v2.json" eval_split="0.2" seed="42" epochs="14" lr="0.08": +train-ai input="dataset/best_moves.jsonl" rl_input="dataset/rl_bootstrap.jsonl" output="models/battlesnake_softmax_v2.json" eval_split="0.2" seed="42" epochs="14" lr="0.08": if [ -f "{{rl_input}}" ]; then python -m server.TrainBattleSnakeAI --input "{{input}}" --input "{{rl_input}}" --output "{{output}}" --eval-split "{{eval_split}}" --seed "{{seed}}" --epochs "{{epochs}}" --lr "{{lr}}"; else python -m server.TrainBattleSnakeAI --input "{{input}}" --output "{{output}}" --eval-split "{{eval_split}}" --seed "{{seed}}" --epochs "{{epochs}}" --lr "{{lr}}"; fi run-trained model="models/battlesnake_softmax_v2.json" port="8000":