Browse Source

fix silly bug

Ali Bellamine 2 years ago
parent
commit
43eaebefd3
1 changed files with 1 additions and 1 deletions
  1. 1 1
      bop_scripts/nn_models.py

+ 1 - 1
bop_scripts/nn_models.py

@@ -214,7 +214,7 @@ class torchMLPClassifier_sklearn (BaseEstimator):
         y_hat_proba_torch = np.concatenate([
             1-y_hat_proba_torch,
             y_hat_proba_torch
-        ], axis=0)
+        ], axis=1)
         y_hat_proba = y_hat_proba_torch
 
         return y_hat_proba