From 6e6b38263ab02d6e9098371c9ef496e3d889e736 Mon Sep 17 00:00:00 2001 From: Eleftheria Tsipidi Date: Fri, 4 Sep 2026 20:36:12 +0200 Subject: [PATCH] Fix constituent parser probabilities indexing --- supar/models/const/crf/parser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/supar/models/const/crf/parser.py b/supar/models/const/crf/parser.py index d1627657..4a57de5d 100644 --- a/supar/models/const/crf/parser.py +++ b/supar/models/const/crf/parser.py @@ -121,7 +121,7 @@ def pred_step(self, batch: Batch) -> Batch: batch.trees = [Tree.build(tree, [(i, j, self.CHART.vocab[label]) for i, j, label in chart]) for tree, chart in zip(trees, chart_preds)] if self.args.prob: - batch.probs = [prob[:i-1, 1:i].cpu() for i, prob in zip(lens, s_span)] + batch.probs = [prob[:i, 1:i+1].cpu() for i, prob in zip(lens, s_span)] return batch @classmethod