update preprocess readme

openreasoner · Nov 13, 2024 · 208ceb2 · 208ceb2
1 parent 75567eb
commit 208ceb2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/preprocess/README.org b/preprocess/README.org
@@ -10,7 +10,7 @@ usage: python cli.py [-h] [-t STEP_TAG] [-s SUFFIX]
               DATASET_TYPE FILE
 
 positional arguments:
-  DATASET_TYPE             Which dataset FILE is (prm800k, math-aps, or math-shepherd)
+  DATASET_TYPE             Which dataset FILE is (prm800k, math-aps, math-aps-tree, or math-shepherd)
   FILE                     Path to the original dataset file
 
 options:
@@ -27,4 +27,5 @@ newlines. Several solutions are suggested in [[https://stackoverflow.com/a/50642
 ** Datasets
 - PRM800K: [[https://github.com/openai/prm800k][GitHub repo]]
 - Math-APS: collected by ~data/gen_data.py~
+- Math-APS-tree: collected by ~data/omegaPRM_v2/~ with ~save_data_tree=True~
 - Math-Shepherd: [[https://huggingface.co/datasets/peiyi9979/Math-Shepherd][Hugging Face dataset]]