From 42b0c213a3758c114ae015e7206244e81943f1a4 Mon Sep 17 00:00:00 2001 From: kapoorlab Date: Sun, 29 Oct 2023 23:39:52 +0100 Subject: [PATCH] covar --- src/napatrackmater/Trackvector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/napatrackmater/Trackvector.py b/src/napatrackmater/Trackvector.py index 338a8bb3..518602d5 100644 --- a/src/napatrackmater/Trackvector.py +++ b/src/napatrackmater/Trackvector.py @@ -691,7 +691,7 @@ def supervised_clustering( ) if not np.isnan(gt_track_array[0]) and shape_dynamic_track_array.shape[0] > 1: - print(gt_track_array[0], shape_dynamic_track_array.shape) + print(gt_track_array[0][0], shape_dynamic_track_array.shape) ( shape_dynamic_covariance, shape_dynamic_eigenvectors, @@ -700,7 +700,7 @@ def supervised_clustering( flattened_covariance = shape_dynamic_covariance.flatten() data_list.append({ 'Flattened_Covariance': flattened_covariance, - 'gt_label': gt_track_array[0] + 'gt_label': gt_track_array[0][0] }) result_dataframe = pd.DataFrame(data_list) if os.path.exists(csv_file_name_original):