哪些功能可以帮助对句子结尾进行分类?序列分类

问题描述 投票:1回答:1

Problem:

我有一对没有句号的句子和他们之间的大写字母。需要将它们彼此分割。我正在寻找一些帮助来选择改进模型的好功能。

Background:

我正在使用pycrfsuite执行序列分类并找到第一个句子的结尾,如下所示:

从棕色语料库中,我将每两句话加在一起,得到他们的pos标签。然后,如果空格跟随它,我用'S'标记句子中的每个标记,如果在句子中跟随它,则标记'P'。然后我删除句子之间的句点,并降低以下标记。我得到这样的东西:

输入:

data = ['I love Harry Potter.', 'It is my favorite book.']

输出:

sent = [('I', 'PRP'), ('love', 'VBP'), ('Harry', 'NNP'), ('Potter', 'NNP'), ('it', 'PRP'), ('is', 'VBZ'), ('my', 'PRP$'), ('favorite', 'JJ'), ('book', 'NN')]
labels = ['S', 'S', 'S', 'P', 'S', 'S', 'S', 'S', 'S']

目前,我提取了这些一般功能:

def word2features2(sent, i):
    word = sent[i][0]
    postag = sent[i][1]

    # Common features for all words
    features = [
        'bias',
        'word.lower=' + word.lower(),
        'word[-3:]=' + word[-3:],
        'word[-2:]=' + word[-2:],
        'word.isupper=%s' % word.isupper(),
        'word.isdigit=%s' % word.isdigit(),
        'postag=' + postag
    ]

    # Features for words that are not
    # at the beginning of a document
    if i > 0:
        word1 = sent[i-1][0]
        postag1 = sent[i-1][1]
        features.extend([
            '-1:word.lower=' + word1.lower(),
            '-1:word.isupper=%s' % word1.isupper(),
            '-1:word.isdigit=%s' % word1.isdigit(),
            '-1:postag=' + postag1
        ])
    else:
        # Indicate that it is the 'beginning of a sentence'
        features.append('BOS')

    # Features for words that are not
    # at the end of a document
    if i < len(sent)-1:
        word1 = sent[i+1][0]
        postag1 = sent[i+1][1]
        features.extend([
            '+1:word.lower=' + word1.lower(),
            '+1:word.isupper=%s' % word1.isupper(),
            '+1:word.isdigit=%s' % word1.isdigit(),
            '+1:postag=' + postag1
        ])
    else:
        # Indicate that it is the 'end of a sentence'
        features.append('EOS')

并使用以下参数训练crf:

    trainer = pycrfsuite.Trainer(verbose=True)

    # Submit training data to the trainer
    for xseq, yseq in zip(X_train, y_train):
        trainer.append(xseq, yseq)

    # Set the parameters of the model
    trainer.set_params({
        # coefficient for L1 penalty
        'c1': 0.1,

        # coefficient for L2 penalty
        'c2': 0.01,

        # maximum number of iterations
        'max_iterations': 200,

        # whether to include transitions that
        # are possible, but not observed
        'feature.possible_transitions': True
    })

    trainer.train('crf.model')

Results:

准确度报告显示:

              precision    recall  f1-score   support

           S       0.99      1.00      0.99    214627
           P       0.81      0.57      0.67      5734

   micro avg       0.99      0.99      0.99    220361
   macro avg       0.90      0.79      0.83    220361
weighted avg       0.98      0.99      0.98    220361

有什么方法可以编辑word2features2()以改进模型? (或任何其他部分)

这是今天完整代码的link

另外,我只是nlp的初学者,所以我非常感谢任何反馈,链接到相关或有用的来源,以及相当简单的解释。非常非常感谢你!

python machine-learning nlp nltk crf
1个回答
1
投票

由于问题的性质,你的课程非常不平衡,我建议使用加权损失,其中P标签的损失比S类的损失更高。我认为问题可能是由于两个类的等效权重,分类器对这些P标签没有给予足够的重视,因为它们对损失的影响非常小。

您可以尝试的另一件事是超参数调整,确保优化宏f1得分,因为无论支持实例的数量如何,它都会给两个类赋予相同的权重。

© www.soinside.com 2019 - 2024. All rights reserved.