TP IA GO -- Nemo D'ACREMONT, Martin EYBEN
Activation de la venv
Exécutez :
python -m venv venv
. ./venv/bin/activate
Télécharger les dépendances
pip -p ./requirements.txt
Chargement des données
Le script télécharge par défaut les données puis les transforme de la façon suivante :
- Duplique les données en faisant des rotations de 90°
- Échange les couleurs des plateaux, en échangeant aussi les scores
- Duplique les données en prenant le symétrique des plateaux
Il les stocke ensuite les datasets sous le nom trainset.pt et testset.pt
dans le dossier output_dir (./current par défaut). Il peut être spécifié
avec l'option -O comme suivant :
mkdir -p /tmp/mydata
python ./main -O /tmp/mydata
Entraînement
Pour entraîner de zéro, il suffit de lancer le script sans spécifier de modèle
avec l'argument -m.
Il est possible de reprendre un entraînement en utilisant l'argument -m,
comme suivant :
python ./main -m ./model.pt
Création des résultats
Afin d'utiliser le modèle pour générer le fichier results.txt, il faut
utiliser l'argument -R avec le chemin du dossier json.gz à utiliser.
Voici un exemple :
python main.py -R positions-to-evaluate-8x8-2025-sample.json.gz -m model.pt