def get_terp_features_on_qats_pair()

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

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

def get_terp_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_TERP_FEATURES' not in globals():
        print('Computing TERp features on all QATS sentence pairs.')
        global QATS_TERP_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)])
        terp_features = get_terp_features(sentences)
        QATS_TERP_FEATURES = {tuple(sentence_pair): features
                              for sentence_pair, features in zip(sentences, terp_features)}
        print('Done.')
    assert (complex_sentence, simple_sentence) in QATS_TERP_FEATURES, 'Sentence pair is not in QATS.'
    return QATS_TERP_FEATURES[(complex_sentence, simple_sentence)]