dataset_info:
features:
- name: chosen
sequence: float64 - name: rejected
sequence: float64
splits: - name: train
num_bytes: 10539475200
num_examples: 160800 - name: test
num_bytes: 560532288
num_examples: 8552
download_size: 8645907100
dataset_size: 11100007488
configs: - config_name: default
data_files:- split: train
path: data/train-* - split: test
path: data/test-*
- split: train
Description
This dataset contains the last-layer-embeddings of the hh-rlhf dataset. The model that we used for embedding is the llama-3-8b pretrained LLM, therefore, the dimension of the embeddings is 4096.