code for ARC Prize 2024
leaderboard for ARC prize
Dataset | Score | Time Taken | cost |
---|---|---|---|
Train | (38/400) 9.5% | 75m | $200 |
- output dimensions not matching
- import errors ('bfs') and missing variables in code
- prefix caching
-
Clone the repository:
git clone https://github.com/yourusername/arcprize.git cd arcprize
-
Install dependencies:
poetry install
-
Set up environment variables: Create a
.env
file in the root directory and add your OpenAI API key:OPENAI_API_KEY=your_openai_api_key
You can run the code for different datasets (train
, eval
, test
) and specify the number of samples to process.
To run predictions on the training dataset:
poetry run python arcprize/main.py --dataset train --n_samples 10
poetry run python arcprize/main.py --dataset eval --n_samples 5
poetry run python arcprize/main.py --dataset test
The main script for running predictions is located in arcprize/main.py
:
The configuration settings are defined in arcprize/config.py
:
Helper functions are implemented in arcprize/helpers.py
: