license:apache-2.0 size_categories:10K<n<100K format:parquet modality:tabular modality:text library:datasets library:pandas library:mlcroissant library:polars region:us
---
license: apache-2.0
dataset_info:
  features:
  - name: question
    dtype: string
  - name: answer
    dtype: string
  - name: target
    dtype: string
  - name: solution_0
    dtype: string
  - name: solution_1
    dtype: string
  - name: solution_2
    dtype: string
  - name: solution_3
    dtype: string
  - name: solution_4
    dtype: string
  - name: correct_1
    dtype: bool
  - name: correct_2
    dtype: bool
  - name: correct_3
    dtype: bool
  - name: correct_4
    dtype: bool
  - name: solution_1_token
    dtype: int64
  - name: solution_2_token
    dtype: int64
  - name: solution_3_token
    dtype: int64
  - name: solution_4_token
    dtype: int64
  - name: __index_level_0__
    dtype: int64
  splits:
  - name: train
    num_bytes: 138779721
    num_examples: 12000
  download_size: 61958370
  dataset_size: 138779721
configs:
- config_name: default
  data_files:
  - split: train
    path: data/train-*
---

## MixChain-Z-PRM12K
MixChain-Z-PRM12K is a dataset containing 12,000 samples (the math split of PRM800K), with each sample containing 5 different solutions. For more details, refer to the paper.
Unlike MixChain-Z-GSM8K, this dataset is unfiltered, meaning that samples with incorrect solutions are retained. We provide extra columns to indicate answer correctness.

### Usage

To load the dataset using the 🤗 datasets library:
```
from datasets import load_dataset

ds = load_dataset("horseee/MixChain-Z-PRM12K")
```

### Dataset Structure
Each sample in the dataset consists of the following columns:
- question: The original question in PRM12K
- answer: The original solution in PRM12K
- target: The original answer from PRM12K
- solution_0: A duplicate column for the column answer
- solution_{1,2,3,4}: Solutions of varying lengths, from shortest to longest (though the length may not strictly increase).
- solution_{1,2,3,4}_token: The token count for each solution.
- correct_{1,2,3,4}: Boolean indicators of whether each solution is correct.