From f0f1416990df7b3c933ccc8d7fc05a84280f4594 Mon Sep 17 00:00:00 2001
From: Paul Best <paul.best@lis-lab.fr>
Date: Fri, 26 May 2023 11:27:54 +0200
Subject: [PATCH] Update file sort_cluster.py
---
new_specie/sort_cluster.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/new_specie/sort_cluster.py b/new_specie/sort_cluster.py
index 3acdcae..56e5dc7 100755
--- a/new_specie/sort_cluster.py
+++ b/new_specie/sort_cluster.py
@@ -33,7 +33,7 @@ parser.add_argument('-min_sample', type=int, default=5, help='Used for HDBSCAN c
parser.add_argument('-eps', type=float, default=0.05, help='Used for HDBSCAN clustering.')
args = parser.parse_args()
-df = pd.read_csv(args.detections, index_col=0)
+df = pd.read_csv(args.detections)
encodings = np.load(args.encodings, allow_pickle=True).item()
idxs, umap = encodings['idx'], encodings['umap']
df.loc[idxs, 'umap_x'] = umap[:,0]
--
GitLab