Skip to content

Commit 30bfc5c

Browse files
committed
correction semi supervised case
1 parent 1669704 commit 30bfc5c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ot/da.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -989,7 +989,7 @@ def fit(self, Xs=None, ys=None, Xt=None, yt=None):
989989

990990
# assumes labeled source samples occupy the first rows
991991
# and labeled target samples occupy the first columns
992-
classes = np.unique(ys)
992+
classes = [c for c in np.unique(ys) if c != -1]
993993
for c in classes:
994994
idx_s = np.where((ys != c) & (ys != -1))
995995
idx_t = np.where(yt == c)

0 commit comments

Comments
 (0)