fix deepIRT loss labels-predictions mismatch

HFUT-LEC · Dec 6, 2023 · 7c11254 · 7c11254
1 parent d0f771c
commit 7c11254
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/edustudio/model/KT/deep_irt.py b/edustudio/model/KT/deep_irt.py
@@ -123,7 +123,7 @@ def predict(self, **kwargs):
             dict: The predictions of the model and the real situation
         """
         y_pd = self(**kwargs)
-        y_pd = y_pd[:, :-1]
+        y_pd = y_pd[:, 1:]
         y_pd = y_pd[kwargs['mask_seq'][:, 1:] == 1]
         y_gt = None
         if kwargs.get('label_seq', None) is not None:
@@ -141,7 +141,7 @@ def get_main_loss(self, **kwargs):
             dict: {'loss_main': loss_value}
         """
         y_pd = self(**kwargs)
-        y_pd = y_pd[:, :-1]
+        y_pd = y_pd[:, 1:]
         y_pd = y_pd[kwargs['mask_seq'][:, 1:] == 1]
         y_gt = kwargs['label_seq'][:, 1:]
         y_gt = y_gt[kwargs['mask_seq'][:, 1:] == 1]