Skip to content

Commit e2c3e92

Browse files
authored
Change SCML supervised docs (#337)
1 parent 44be909 commit e2c3e92

File tree

1 file changed

+12
-7
lines changed

1 file changed

+12
-7
lines changed

metric_learn/scml.py

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -473,13 +473,18 @@ class SCML_Supervised(_BaseSCML, TransformerMixin):
473473
474474
Examples
475475
--------
476-
>>> from metric_learn import SCML
477-
>>> triplets = np.array([[[1.2, 3.2], [2.3, 5.5], [2.1, 0.6]],
478-
>>> [[4.5, 2.3], [2.1, 2.3], [7.3, 3.4]]])
479-
>>> scml = SCML(random_state=42)
480-
>>> scml.fit(triplets)
481-
SCML(beta=1e-5, B=None, max_iter=100000, verbose=False,
482-
preprocessor=None, random_state=None)
476+
>>> from metric_learn import SCML_Supervised
477+
>>> from sklearn.datasets import load_iris
478+
>>> iris_data = load_iris()
479+
>>> X = iris_data['data']
480+
>>> Y = iris_data['target']
481+
>>> scml = SCML_Supervised(random_state=33)
482+
>>> scml.fit(X, Y)
483+
SCML_Supervised(random_state=33)
484+
>>> scml.score_pairs([[X[0], X[1]], [X[0], X[2]]])
485+
array([1.84640733, 1.55984363])
486+
>>> scml.get_metric()(X[0], X[1])
487+
1.8464073327922157
483488
484489
References
485490
----------

0 commit comments

Comments
 (0)