Merge pull request #13 from badranX/qikt

fix QIKT response data leakage
HFUT-LEC · Nov 25, 2023 · ea1f688 · ea1f688
2 parents e92b7d1 + c5460c3
commit ea1f688
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/edustudio/model/KT/qikt.py b/edustudio/model/KT/qikt.py
@@ -71,12 +71,9 @@ def forward(self, exer_seq, label_seq, cpt_seq, cpt_seq_mask, **kwargs):
                           dim=2) / cpt_seq_mask.sum(dim=2).unsqueeze(2).repeat(1, 1, self.modeltpl_cfg['emb_size'])
 
 
-        e_tmp = torch.cat((emb_q, emb_c), dim=2)
-        mask_e = label_seq.unsqueeze(-1).repeat(1, 1, e_tmp.shape[-1]).to(torch.float)
-        emb_qca = torch.cat((mask_e*e_tmp, (1-mask_e)*e_tmp), dim=-1)
-
-        mask_c = label_seq.unsqueeze(-1).repeat(1, 1, emb_c.shape[-1]).to(torch.float)
-        emb_qc = torch.cat((mask_c * emb_c, (1 - mask_c) * emb_c), dim=-1)
+        emb_qc = torch.cat((emb_q, emb_c), dim=2)
+        mask_e = label_seq.unsqueeze(-1).repeat(1, 1, emb_qc.shape[-1]).to(torch.float)
+        emb_qca = torch.cat((mask_e*emb_qc, (1-mask_e)*emb_qc), dim=-1)
 
         emb_qc_shift = emb_qc[:, 1:, :]
         emb_qca_current = emb_qca[:, :-1, :]