From 35e3828f9f4697246efd4113197a3ee838a0c3b6 Mon Sep 17 00:00:00 2001
From: Maja Jablonska <majajjablonska@gmail.com>
Date: Mon, 15 Jan 2024 11:52:22 +0100
Subject: [PATCH] debug messages
---
combo/combo_model.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/combo/combo_model.py b/combo/combo_model.py
index b87eb1c..3ab90c8 100644
--- a/combo/combo_model.py
+++ b/combo/combo_model.py
@@ -173,6 +173,7 @@ class ComboModel(Model, FromParameters):
mapped_gold_labels = []
for _, cat_indices in self.morphological_feat.slices.items():
try:
+ print("Feats dimensions: ", feats.shape, "cat_indices shape", len(cat_indices), "min and max", min(cat_indices), max(cat_indices))
mapped_gold_labels.append(feats[:, :, cat_indices].argmax(dim=-1))
except TypeError:
raise ConfigurationError('Feats is None - if no feats are provided, the morphological_feat property should be set to None.')
--
GitLab