Merge pull request #3491 from sheldon-roberts/master

helpmefindaname · web-flow · commit 355e2c0b9f00 · 2024-07-04T16:58:25.000+02:00
Fix TextRegressor label_name bug
diff --git a/flair/models/text_regression_model.py b/flair/models/text_regression_model.py
@@ -64,7 +64,8 @@ def forward_loss(self, sentences: List[Sentence]) -> Tuple[torch.Tensor, int]:
 
     def _labels_to_tensor(self, sentences: List[Sentence]):
         indices = [
-            torch.tensor([float(label.value) for label in sentence.labels], dtype=torch.float) for sentence in sentences
+            torch.tensor([float(label.value) for label in sentence.get_labels(self.label_name)], dtype=torch.float)
+            for sentence in sentences
         ]
 
         vec = torch.cat(indices, 0).to(flair.device)

Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,8 @@ def forward_loss(self, sentences: List[Sentence]) -> Tuple[torch.Tensor, int]:`
`64`	`64`
`65`	`65`	`def _labels_to_tensor(self, sentences: List[Sentence]):`
`66`	`66`	`indices = [`
`67`		`- torch.tensor([float(label.value) for label in sentence.labels], dtype=torch.float) for sentence in sentences`
	`67`	`+ torch.tensor([float(label.value) for label in sentence.get_labels(self.label_name)], dtype=torch.float)`
	`68`	`+ for sentence in sentences`
`68`	`69`	`]`
`69`	`70`
`70`	`71`	`vec = torch.cat(indices, 0).to(flair.device)`