Transpose confusion matrix in theta initialisation with pretrain

arthurPignet · arthurPignet · commit c1597867ebaa · 2021-01-06T16:58:15.000+01:00
Signed-off-by: arthurPignet &lt;arthur.pignet@mines-paristech.fr&gt;

Signed-off-by: arthurPignet &lt;arthur.pignet@mines-paristech.fr&gt;
diff --git a/mplc/multi_partner_learning.py b/mplc/multi_partner_learning.py
@@ -546,7 +546,7 @@ def fit(self):
                 confusion = confusion_matrix(np.argmax(p.y_train, axis=1),
                                              np.argmax(pretrain_model.predict(p.x_train), axis=1),
                                              normalize='pred')
-                p.noise_layer_weights = [np.log(confusion + 1e-8)]
+                p.noise_layer_weights = [np.log(confusion.T + 1e-8)]
             self.model_weights[:-1] = self.pretrain_mpl.model_weights[:-1]
         else:
             for p in self.partners_list:
