pyrit.datasets.fetch_wmdp_dataset

Contents

pyrit.datasets.fetch_wmdp_dataset#

fetch_wmdp_dataset(category: str | None = None) → QuestionAnsweringDataset[source]#

Fetch WMDP examples and create a QuestionAnsweringDataset.

Parameters:: category (str) – The dataset category, one of “cyber”, “bio”, “chem”
Returns:: A QuestionAnsweringDataset containing the examples.
Return type:: QuestionAnsweringDataset

Note

For more information and access to the original dataset and related materials, visit: https://huggingface.co/datasets/cais/wmdp