Skip to content
Snippets Groups Projects
Commit f0f14169 authored by Paul Best's avatar Paul Best
Browse files

Update file sort_cluster.py

parent 02b15b31
Branches
No related tags found
No related merge requests found
......@@ -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]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment