Skip to content

Commit eff278e

Browse files
author
William de Vazelhes
committed
REF rename n_features_out to num_dims to be more coherent with current algorithms
1 parent 12b5429 commit eff278e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

metric_learn/base_metric.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ class MahalanobisMixin(six.with_metaclass(ABCMeta, BaseMetricLearner)):
6666
6767
Attributes
6868
----------
69-
transformer_ : `np.ndarray`, shape=(n_features_out, n_features)
69+
transformer_ : `np.ndarray`, shape=(num_dims, n_features)
7070
The learned linear transformation ``L``.
7171
"""
7272

@@ -110,7 +110,7 @@ def embed(self, X):
110110
111111
Returns
112112
-------
113-
X_embedded : `numpy.ndarray`, shape=(n_samples, n_features_out)
113+
X_embedded : `numpy.ndarray`, shape=(n_samples, num_dims)
114114
The embedded data points.
115115
"""
116116
return X.dot(self.transformer_.T)

0 commit comments

Comments
 (0)