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: 60714653
num_examples: 42537 - name: test
num_bytes: 3420426
num_examples: 2312
download_size: 35346234
dataset_size: 64135079
configs: - config_name: default
data_files:- split: train
path: data/train-* - split: test
path: data/test-*
- split: train
TRL's Anthropic HH Dataset
Use the code from https://huggingface.co/datasets/trl-internal-testing/hh-rlhf-helpful-base-trl-style,
to preprocess load_dataset("Anthropic/hh-rlhf", data_dir="harmless-base")
in the TRL style