dataset_info:
features:
- name: chosen
list:- name: content
dtype: string - name: role
dtype: string
- name: content
- name: rejected
list:- name: content
dtype: string - name: role
dtype: string
- name: content
- name: prompt
dtype: string
splits: - name: train
num_bytes: 71553768
num_examples: 43835 - name: test
num_bytes: 3858452
num_examples: 2354
download_size: 42037894
dataset_size: 75412220
configs: - config_name: default
data_files:- split: train
path: data/train-* - split: test
path: data/test-*
- split: train
TRL's Anthropic HH Dataset
We preprocess the dataset using our standard prompt, chosen, rejected format.
Reproduce this dataset
- Download the
anthropic_hh.pyfrom the https://huggingface.co/datasets/trl-internal-testing/hh-rlhf-helpful-base-trl-style/tree/0.1.0. - Run
python examples/datasets/anthropic_hh.py --push_to_hub --hf_entity trl-internal-testing