This dataset contains the complete Graph-R1 graph reasoning dataset with all difficulty levels (1-5).

train_graph_all_levels.parquet: Combined training data from all levels (with level column) test_graph_and_math_all_levels.parquet: Combined test data from all levels (with level column)
test_graph_mixedsize_cleaned.parquet: Mixed size test data (with level='mixed')

import pandas as pd