fix a bug for MFI

PranavUikey · Oct 28, 2022 · a079721 · a079721
1 parent 43caf07
commit a079721
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/CAT/strategy/KLI_strategy.py b/CAT/strategy/KLI_strategy.py
@@ -34,7 +34,7 @@ class MKLIStrategy(KLIStrategy):
 
     def __init__(self):
         super().__init__()
-    
+
     @property
     def name(self):
         return 'Multivariate Kullback-Leibler Information Strategy'
diff --git a/CAT/strategy/MFI_strategy.py b/CAT/strategy/MFI_strategy.py
@@ -26,7 +26,7 @@ def adaptest_select(self, model: AbstractModel, adaptest_data: AdapTestDataset):
             'the models must implement get_pred method for accelerating'
         pred_all = model.get_pred(adaptest_data)
         if self.I is None:
-            self.I = [np.zeros((model.model.num_dim, model.model.num_dim))] * adaptest_data.num_students    
+            self.I = [np.zeros((model.model.num_dim, model.model.num_dim)) for _ in range(adaptest_data.num_students)]
         selection = {}
         n = len(adaptest_data.tested[0])
         for sid in range(adaptest_data.num_students):
@@ -41,11 +41,11 @@ def adaptest_select(self, model: AbstractModel, adaptest_data: AdapTestDataset):
             selection[sid] = untested_questions[j]
             self.I[sid] += untested_fisher[j]
         return selection
-    
+
 class DoptStrategy(MFIStrategy):
     def __init__(self):
         super().__init__()
-    
+
     @property
     def name(self):
         return 'D-Optimality Strategy'