from datasets.features import ClassLabel, Features, Value
from datasets.load import load_dataset


def load_sensai_dataset():
    dataset = load_dataset(
        "holodata/sensai",
        features=Features({
            "body": Value("string"),
            "toxic": ClassLabel(num_classes=2, names=['0', '1'])
        }))
    dataset = dataset['train']
    return dataset