sklearn.neighbors
.NeighborhoodComponentsAnalysis¶

class
sklearn.neighbors.
NeighborhoodComponentsAnalysis
(n_components=None, *, init='auto', warm_start=False, max_iter=50, tol=1e05, callback=None, verbose=0, random_state=None)[source]¶ Neighborhood Components Analysis
Neighborhood Component Analysis (NCA) is a machine learning algorithm for metric learning. It learns a linear transformation in a supervised fashion to improve the classification accuracy of a stochastic nearest neighbors rule in the transformed space.
Read more in the User Guide.
 Parameters
 n_componentsint, default=None
Preferred dimensionality of the projected space. If None it will be set to
n_features
. init{‘auto’, ‘pca’, ‘lda’, ‘identity’, ‘random’} or ndarray of shape (n_features_a, n_features_b), default=’auto’
Initialization of the linear transformation. Possible options are ‘auto’, ‘pca’, ‘lda’, ‘identity’, ‘random’, and a numpy array of shape (n_features_a, n_features_b).
 ‘auto’
Depending on
n_components
, the most reasonable initialization will be chosen. Ifn_components <= n_classes
we use ‘lda’, as it uses labels information. If not, butn_components < min(n_features, n_samples)
, we use ‘pca’, as it projects data in meaningful directions (those of higher variance). Otherwise, we just use ‘identity’. ‘pca’
n_components
principal components of the inputs passed tofit
will be used to initialize the transformation. (SeePCA
) ‘lda’
min(n_components, n_classes)
most discriminative components of the inputs passed tofit
will be used to initialize the transformation. (Ifn_components > n_classes
, the rest of the components will be zero.) (SeeLinearDiscriminantAnalysis
) ‘identity’
If
n_components
is strictly smaller than the dimensionality of the inputs passed tofit
, the identity matrix will be truncated to the firstn_components
rows. ‘random’
The initial transformation will be a random array of shape
(n_components, n_features)
. Each value is sampled from the standard normal distribution. numpy array
n_features_b must match the dimensionality of the inputs passed to
fit
and n_features_a must be less than or equal to that. Ifn_components
is not None, n_features_a must match it.
 warm_startbool, default=False
If True and
fit
has been called before, the solution of the previous call tofit
is used as the initial linear transformation (n_components
andinit
will be ignored). max_iterint, default=50
Maximum number of iterations in the optimization.
 tolfloat, default=1e5
Convergence tolerance for the optimization.
 callbackcallable, default=None
If not None, this function is called after every iteration of the optimizer, taking as arguments the current solution (flattened transformation matrix) and the number of iterations. This might be useful in case one wants to examine or store the transformation found after each iteration.
 verboseint, default=0
If 0, no progress messages will be printed. If 1, progress messages will be printed to stdout. If > 1, progress messages will be printed and the
disp
parameter ofscipy.optimize.minimize
will be set toverbose  2
. random_stateint or numpy.RandomState, default=None
A pseudo random number generator object or a seed for it if int. If
init='random'
,random_state
is used to initialize the random transformation. Ifinit='pca'
,random_state
is passed as an argument to PCA when initializing the transformation. Pass an int for reproducible results across multiple function calls. See :term:Glossary <random_state>
.
 Attributes
 components_ndarray of shape (n_components, n_features)
The linear transformation learned during fitting.
 n_iter_int
Counts the number of iterations performed by the optimizer.
 random_state_numpy.RandomState
Pseudo random number generator object used during initialization.
References
 1
J. Goldberger, G. Hinton, S. Roweis, R. Salakhutdinov. “Neighbourhood Components Analysis”. Advances in Neural Information Processing Systems. 17, 513520, 2005. http://www.cs.nyu.edu/~roweis/papers/ncanips.pdf
 2
Wikipedia entry on Neighborhood Components Analysis https://en.wikipedia.org/wiki/Neighbourhood_components_analysis
Examples
>>> from sklearn.neighbors import NeighborhoodComponentsAnalysis >>> from sklearn.neighbors import KNeighborsClassifier >>> from sklearn.datasets import load_iris >>> from sklearn.model_selection import train_test_split >>> X, y = load_iris(return_X_y=True) >>> X_train, X_test, y_train, y_test = train_test_split(X, y, ... stratify=y, test_size=0.7, random_state=42) >>> nca = NeighborhoodComponentsAnalysis(random_state=42) >>> nca.fit(X_train, y_train) NeighborhoodComponentsAnalysis(...) >>> knn = KNeighborsClassifier(n_neighbors=3) >>> knn.fit(X_train, y_train) KNeighborsClassifier(...) >>> print(knn.score(X_test, y_test)) 0.933333... >>> knn.fit(nca.transform(X_train), y_train) KNeighborsClassifier(...) >>> print(knn.score(nca.transform(X_test), y_test)) 0.961904...
Methods
fit
(X, y)Fit the model according to the given training data.
fit_transform
(X[, y])Fit to data, then transform it.
get_params
([deep])Get parameters for this estimator.
set_params
(**params)Set the parameters of this estimator.
transform
(X)Applies the learned transformation to the given data.

fit
(X, y)[source]¶ Fit the model according to the given training data.
 Parameters
 Xarraylike of shape (n_samples, n_features)
The training samples.
 yarraylike of shape (n_samples,)
The corresponding training labels.
 Returns
 selfobject
returns a trained NeighborhoodComponentsAnalysis model.

fit_transform
(X, y=None, **fit_params)[source]¶ Fit to data, then transform it.
Fits transformer to
X
andy
with optional parametersfit_params
and returns a transformed version ofX
. Parameters
 Xarraylike of shape (n_samples, n_features)
Input samples.
 yarraylike of shape (n_samples,) or (n_samples, n_outputs), default=None
Target values (None for unsupervised transformations).
 **fit_paramsdict
Additional fit parameters.
 Returns
 X_newndarray array of shape (n_samples, n_features_new)
Transformed array.

get_params
(deep=True)[source]¶ Get parameters for this estimator.
 Parameters
 deepbool, default=True
If True, will return the parameters for this estimator and contained subobjects that are estimators.
 Returns
 paramsdict
Parameter names mapped to their values.

set_params
(**params)[source]¶ Set the parameters of this estimator.
The method works on simple estimators as well as on nested objects (such as
Pipeline
). The latter have parameters of the form<component>__<parameter>
so that it’s possible to update each component of a nested object. Parameters
 **paramsdict
Estimator parameters.
 Returns
 selfestimator instance
Estimator instance.