Fix issue with encoder.py and Advanced Space

PonteIneptique · PonteIneptique · commit 0ea9830dd96f · 2022-04-19T20:47:08.000+02:00
diff --git a/boudams/encoder.py b/boudams/encoder.py
@@ -277,12 +277,12 @@ def format_confusion_matrix(self, confusion: List[List[int]]):
             for head, confusion_row in zip(header[1:], confusion)
         ]
 
-        col_pad = header.index(self.pad_token)
+        col_pad = header.index(self.mode.index_to_masks_name.get(self.pad_token_index))
         return tabulate.tabulate(
             [
                 row[:col_pad] + row[col_pad+1:]
                 for (col_id, row) in enumerate(confusion)
-                if row[0] != self.pad_token
+                if row[0] != "PAD"
             ],
-            headers=[col for col in header if col != self.pad_token]
+            headers=[col for col in header if col != "PAD"]
         )

Original file line number	Diff line number	Diff line change
`@@ -277,12 +277,12 @@ def format_confusion_matrix(self, confusion: List[List[int]]):`
`277`	`277`	`for head, confusion_row in zip(header[1:], confusion)`
`278`	`278`	`]`
`279`	`279`
`280`		`- col_pad = header.index(self.pad_token)`
	`280`	`+ col_pad = header.index(self.mode.index_to_masks_name.get(self.pad_token_index))`
`281`	`281`	`return tabulate.tabulate(`
`282`	`282`	`[`
`283`	`283`	`row[:col_pad] + row[col_pad+1:]`
`284`	`284`	`for (col_id, row) in enumerate(confusion)`
`285`		`- if row[0] != self.pad_token`
	`285`	`+ if row[0] != "PAD"`
`286`	`286`	`],`
`287`		`- headers=[col for col in header if col != self.pad_token]`
	`287`	`+ headers=[col for col in header if col != "PAD"]`
`288`	`288`	`)`