strong_reject.load_datasets#

Load the StrongREJECT dataset.

strong_reject.load_datasets.load_strongreject() Dataset#

Load the full StrongREJECT dataset.

Returns:

Full StrongREJECT dataset.

Return type:

Dataset

strong_reject.load_datasets.load_strongreject_small() Dataset#

Load the StrongREJECT small dataset.

Returns:

StrongREJECT small.

Return type:

Dataset

strong_reject.load_datasets.load_wmdp_open_ended() Dataset#

Load an open-ended version of the WMDP chem and bio datasets.

Returns:

Open-ended version of WMDP chem and bio.

Return type:

Dataset