This repository contains a script for generating Supervised Fine-Tuning (SFT) datasets from ARC-AGI puzzles with data augmentation.
Overview
The re-arc-2-sft-generator-notebook.py script takes original ARC-AGI training puzzles and creates 40 augmented variations of each puzzle through geometric transformations and color permutations. This significantly expands the training dataset to improve model performance.
Puzzle ID Naming… See the full description on the dataset page: https://huggingface.co/datasets/Slyracoon23/arc-agi-sft-dataset-augmented.