oasst1/README.md
2023-04-15 14:27:59 +00:00

5.9 KiB

license dataset_info language tags size_categories pretty_name
apache-2.0
features splits download_size dataset_size
name dtype
message_id string
name dtype
parent_id string
name dtype
user_id string
name dtype
created_date string
name dtype
text string
name dtype
role string
name dtype
lang string
name dtype
review_count int32
name dtype
review_result bool
name dtype
deleted bool
name dtype
rank int32
name dtype
synthetic bool
name dtype
model_name string
name struct
detoxify
name dtype
toxicity float64
name dtype
severe_toxicity float64
name dtype
obscene float64
name dtype
identity_attack float64
name dtype
insult float64
name dtype
threat float64
name dtype
sexual_explicit float64
name dtype
message_tree_id string
name dtype
tree_state string
name sequence
emojis
name dtype
name string
name dtype
count int32
name sequence
labels
name dtype
name string
name dtype
value float64
name dtype
count int32
name num_bytes num_examples
train 100367999 84437
name num_bytes num_examples
validation 5243405 4401
41596430 105611404
en
es
ru
de
pl
th
vi
sv
bn
da
he
it
fa
sk
id
nb
el
nl
hu
eu
zh
eo
ja
ca
cs
bg
fi
pt
tr
ro
ar
uk
gl
fr
ko
human-feedback
10K<n<100K
OpenAssistant Conversations

OpenAssistant Conversations Dataset (OASST1)

Dataset Description

Dataset Summary

In an effort to democratize research on large-scale alignment, we release OpenAssistant Conversations (OASST1), a human-generated, human-annotated assistant-style conversation corpus consisting of 161,443 messages distributed across 66,497 conversation trees, in 35 different languages, annotated with 461,292 quality ratings. The corpus is a product of a worldwide crowd-sourcing effort involving over 13,500 volunteers.

Please our paper for further details.

Dataset Structure

This dataset contains demonstrations of human-assistant conversations which were collected on the open-assistant.io website until April, 12 2023.

Conversations are exported as conversation trees with messages as nodes. The root node of a conversation tree is called the initial prompt. Each message can have multiple replies. Nodes with more than one reply can have a rank field indicating the user preference (the most preferred message has rank 0).

All messages have a role which can either be "assistant" or "prompter". The roles in conversation threads from prompt to leaf node in a conversation tree are stricly alternating between "assistant" and "prompter".

Please refer to oasst-data for details about the data structure and python code to read and write jsonl files containing oasst objects.

Main Dataset Files

Data is provided either as nested messages in conversation trees (extension .trees.jsonl.gz) or as flat list of messages (extension .messages.jsonl.gz).

Full conversation trees can be reconstructed from flat messages using the parent_id and message_id properties to identify their parent-child relationship. The message_tree_id and tree_state properties (only present in flat messages files) can be used to find all all messages of a message tree or to select trees by their state.

Ready For Export Trees

2023-04-12_oasst_ready.trees.jsonl.gz       10364 trees with 88838 total messages
2023-04-12_oasst_ready.messages.jsonl.gz    88838 messages

Trees in ready_for_export state without spam and deleted messages including message labels. The oasst_ready-trees file is normally sufficient for supervised fine-tuning (SFT) & reward model (RM) training.

All Trees

2023-04-12_oasst_all.trees.jsonl.gz         66497 trees with 161443 total messages
2023-04-12_oasst_all.messages.jsonl.gz     161443 messages

All trees including those in states prompt_lottery_waiting, aborted_low_grade, halted_by_moderator.

Supplemental Exports: Spam & Prompts

2023-04-12_oasst_spam.messages.jsonl.gz

Messages which were deleted or have a negative review result ("review_result": false). Beside low quality a frequent reason for message deletion is a wrong language tag.

2023-04-12_oasst_prompts.messages.jsonl.gz

All non-deleted initial prompt messages with positile spam review result of trees in ready_for_export or prompt_lottery_waiting state.

Languages

OpenAssistant Conversations incorporates 35 different languages with a distribution of messages as follows:

Languages with over 1000 messages

  • English: 71956
  • Spanish: 43061
  • Russian: 9089
  • German: 5279
  • Chinese: 4962
  • French: 4251
  • Thai: 3042
  • Portuguese (Brazil): 2969
  • Catalan: 2260
  • Korean: 1553
  • Ukrainian: 1352
  • Italian: 1320
  • Japanese: 1018
Languages with under 1000 messages
  • Vietnamese: 952
  • Basque: 947
  • Polish: 886
  • Hungarian: 811
  • Arabic: 666
  • Dutch: 628
  • Swedish: 512
  • Turkish: 454
  • Finnish: 386
  • Czech: 372
  • Danish: 358
  • Galician: 339
  • Hebrew: 255
  • Romanian: 200
  • Norwegian Bokmål: 133
  • Indonesian: 115
  • Bulgarian: 95
  • Bengali: 82
  • Persian: 72
  • Greek: 66
  • Esperanto: 59
  • Slovak: 19