size_categories:100K<n<1M format:parquet library:datasets library:dask library:mlcroissant library:polars region:us
---
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-*
---

## 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.