add Training for AI and AI Model and allow to collect rl data from BestBattleSnake

justfile

@@ -159,7 +159,7 @@ test-local-4 mode="standard" map="standard" base_port="9101" snake="BestBattleSn
     -g "{{mode}}" --map "{{map}}" --seed "{{seed}}" $BROWSER_FLAG
 
 # ------------------------------------------------------------------------------
-#  Fataset helpers
+#  Dataset helpers
 # ------------------------------------------------------------------------------
 
 export-dataset input="data" output="data/dataset/good_moves.jsonl":
@@ -170,3 +170,9 @@ 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":
+  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":
+  TRAINED_SNAKE_MODEL="{{model}}" SNAKE="TrainedBattleSnake" PORT="{{port}}" "{{justfile_directory()}}/main.py"