Skip to content

Commit b4bdec4

Browse files
author
William de Vazelhes
committed
update docstrings with change for +1/-1 labels (see #92 (comment))
1 parent 374a851 commit b4bdec4

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

metric_learn/itml.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def fit(self, pairs, y, bounds=None):
8787
pairs: array-like, shape=(n_constraints, 2, n_features)
8888
Array of pairs. Each row corresponds to two points.
8989
y: array-like, of shape (n_constraints,)
90-
Labels of constraints. Should be 0 for dissimilar pair, 1 for similar.
90+
Labels of constraints. Should be -1 for dissimilar pair, 1 for similar.
9191
bounds : list (pos,neg) pairs, optional
9292
bounds on similarity, s.t. d(X[a],X[b]) < pos and d(X[c],X[d]) > neg
9393

metric_learn/mmc.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def fit(self, pairs, y):
6767
pairs: array-like, shape=(n_constraints, 2, n_features)
6868
Array of pairs. Each row corresponds to two points.
6969
y: array-like, of shape (n_constraints,)
70-
Labels of constraints. Should be 0 for dissimilar pair, 1 for similar.
70+
Labels of constraints. Should be -1 for dissimilar pair, 1 for similar.
7171
7272
Returns
7373
-------

metric_learn/sdml.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def fit(self, pairs, y):
6565
pairs: array-like, shape=(n_constraints, 2, n_features)
6666
Array of pairs. Each row corresponds to two points.
6767
y: array-like, of shape (n_constraints,)
68-
Labels of constraints. Should be 0 for dissimilar pair, 1 for similar.
68+
Labels of constraints. Should be -1 for dissimilar pair, 1 for similar.
6969
7070
Returns
7171
-------

0 commit comments

Comments
 (0)