diff --git a/new_specie/sort_cluster.py b/new_specie/sort_cluster.py index 3acdcaeb72ecb60f4fcce87c34e64fa74d3ba125..56e5dc7705bf92248f9a53b9296fdd392b9646e6 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]