fix bug with concetenation axis

BIMSBbioinfo · Dec 14, 2023 · 0f5a24b · 0f5a24b
1 parent 0df75e7
commit 0f5a24b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flexynesis/utils.py b/flexynesis/utils.py
@@ -180,7 +180,7 @@ def evaluate_wrapper(y_pred_dict, dataset):
 def evaluate_baseline_performance(train_dataset, test_dataset, variable_name, n_folds=5, n_jobs = 4):
     def prepare_data(data_object):
         # Concatenate Data Matrices
-        X = np.concatenate([tensor for tensor in data_object.dat.values()], axis=0)
+        X = np.concatenate([tensor for tensor in data_object.dat.values()], axis=1)
 
         # Prepare Labels
         y = np.array(data_object.ann[variable_name])