diff --git a/dan/predict/prediction.py b/dan/predict/prediction.py index 8d021712c2d36d049566f81ebbee5f78021a4e16..fa4bcea0ffa3cf434125fed4f913cbad171e3d4d 100644 --- a/dan/predict/prediction.py +++ b/dan/predict/prediction.py @@ -350,7 +350,7 @@ def run( index = [pos for pos, char in enumerate(text) if char in ["â“", "â“Ÿ", "â““", "â“¡"]] # calculates scores by token - score_by_token = [ + result["confidences"]["by ner token"] = [ { "text": f"{text[current: next_token-1]}", "confidence_ner": f"{np.around(np.mean(char_confidences[current : next_token-1]), 2)}", @@ -359,9 +359,6 @@ def run( ] result["confidences"]["total"] = np.around(np.mean(char_confidences), 2) - result["confidences"]["by ner token"] = [] - for entity in score_by_token: - result["confidences"]["by ner token"].append(entity) for level in confidence_score_levels: result["confidences"][level] = []