ThtratLandscapeChat / datasets_loader.py
S-Dreamer's picture
Update datasets_loader.py
1c34bfc verified
raw
history blame
247 Bytes
from datasets import load_dataset
def load_redteamer_mistral():
return load_dataset("romaingrx/red-teamer-mistral-", split="train")
def load_multi_class_redteaming():
return load_dataset("SummerSigh/Muti-Class-Redteaming", split="train")