File size: 418 Bytes
5ce0b20
 
cdc442b
 
 
5ce0b20
 
1
2
3
4
5
6
7
test_cases = [
    {
        "predictions": ['[Hotel New Home, Hopeland]'],
        "references": [['Bar Guisness', 'Hotel New Home', 'New Hopeland']],
        "result": {'bert_score_precision': 0.8470218181610107, 'bert_score_recall': 0.9131535291671753, 'bert_score_f1': 0.8788453936576843, 'bleu-1': 0.5714285714285714, 'precision': [6.0], 'rappel': [15.0], 'macro-mean': [10.5], 'median macro-mean': 10.5}
    },
]