Merge pull request #311 from SubstraFoundation/fix-smodel-init-with-pretrain

Fix smodel

Author: arthurPignet

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: