Home Explore Help
Sign In
alibell
/
biologyOrderPredictor
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

fix silly bug

Ali Bellamine 2 years ago
parent
8dd43b0f9a
commit
43eaebefd3
1 changed files with 1 additions and 1 deletions
Split View Show Diff Stats
  1. 1 1
      bop_scripts/nn_models.py

+ 1 - 1
bop_scripts/nn_models.py
View File 

@@ -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