0

我需要对情绪标签中的一些文本进行分类。我正在使用多标签分类，因为相同的文本可以包含不止一种情绪，但我想实现其中一些是不相交的，比如快乐/悲伤或平静/愤怒。

假设我在 Python 中有这段代码：

from simpletransformers.classification import (
    MultiLabelClassificationModel, MultiLabelClassificationArgs
)

model_args = MultiLabelClassificationArgs(num_train_epochs=1)

# Create a MultiLabelClassificationModel
model = MultiLabelClassificationModel(
    "roberta", "roberta-base", num_labels=4,
)

使用此示例：

train_data = [
    ["AAA", [1, 0, 0, 1]],
    ["BBB", [0, 1, 1, 1]],
    ["CCC", [1, 0, 1, 1]],
]

而且我想设置第一个和第二个标签必须是不相交的。我怎么能做到？

4

0 回答 0