Skip to content

Commit 3a381bb

Browse files
committed
fix: X_pool masked out in pairwise_distances_argmin_min call
1 parent 38cf14c commit 3a381bb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modAL/batch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def select_instance(
9090
# Compute pairwise distance (and then similarity) scores from every unlabeled record
9191
# to every record in X_training. The result is an array of shape (n_samples, ).
9292
if n_jobs == 1 or n_jobs is None:
93-
_, distance_scores = pairwise_distances_argmin_min(X_pool, X_training, metric=metric)
93+
_, distance_scores = pairwise_distances_argmin_min(X_pool[mask], X_training, metric=metric)
9494
else:
9595
distance_scores = pairwise_distances(X_pool[mask], X_training, metric=metric, n_jobs=n_jobs).min(axis=1)
9696

0 commit comments

Comments
 (0)