task 1 correction

This commit is contained in:
Niraj Sonje 2023-11-28 13:40:12 -07:00
parent af0f50d981
commit 1a07e0c957

View File

@ -277,7 +277,7 @@ def calculate_label_representatives(fd_collection, label, feature_model):
img_fds[feature_model]
).flatten() # get the specific feature model's feature vector
for img_fds in fd_collection.find(
{"true_label": label, "$mod": [2,0]}
{"true_label": label, "image_id": {"$mod": [2,0]}}
) # repeat for all images
]