CollieDatasetForClassification

class CollieDatasetForClassification(dataset, tokenizer=None, add_special_tokens=True, shuffle=False, seed=1024, max_length=-1, style='harness')

CoLLie 中的分类任务数据集 须搭配 EvaluatorForClassfication 使用。需提供的数据格式形似:

[
    {
        "input": "这是prompt部分的文本",
        "output": ["类别1", "类别2", "类别3"],
        "target": 0
    },
    ...
]