def get_quest_features_on_qats_pair()

in tseval/evaluation/quest.py [0:0]

• 14 lines of code
• 3 McCabe index (conditional complexity)

def get_quest_features_on_qats_pair(complex_sentence, simple_sentence):
    # Computing features on a single sentence pair is as long as computing all sentence pairs in QATS at once
    if 'QATS_QUEST_FEATURES' not in globals():
        print('Computing QuEst features on all QATS sentence pairs.')
        global QATS_QUEST_FEATURES
        train_sentences, _ = get_qats_train_data('simplicity')
        test_sentences, _ = get_qats_test_data('simplicity')
        sentences = np.concatenate([train_sentences, test_sentences])
        sentences = np.concatenate([sentences, np.flip(sentences, axis=1)])
        quest_features = get_quest_features(sentences)
        QATS_QUEST_FEATURES = {tuple(sentence_pair): features
                               for sentence_pair, features in zip(sentences, quest_features)}
        print('Done.')
    assert (complex_sentence, simple_sentence) in QATS_QUEST_FEATURES, 'Sentence pair is not in QATS.'
    return QATS_QUEST_FEATURES[(complex_sentence, simple_sentence)]