Skip to content

Commit d2b9c80

Browse files
rand0wnjeremiedbb
andauthored
MAINT Parameters validation for chi2_kernel (scikit-learn#26153)
Co-authored-by: jeremiedbb <jeremiedbb@yahoo.fr>
1 parent e490da9 commit d2b9c80

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

sklearn/metrics/pairwise.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1678,6 +1678,14 @@ def additive_chi2_kernel(X, Y=None):
16781678
return result
16791679

16801680

1681+
@validate_params(
1682+
{
1683+
"X": ["array-like"],
1684+
"Y": ["array-like", None],
1685+
"gamma": [Interval(Real, 0, None, closed="neither"), Hidden(np.ndarray)],
1686+
},
1687+
prefer_skip_nested_validation=True,
1688+
)
16811689
def chi2_kernel(X, Y=None, gamma=1.0):
16821690
"""Compute the exponential chi-squared kernel between X and Y.
16831691
@@ -1698,7 +1706,7 @@ def chi2_kernel(X, Y=None, gamma=1.0):
16981706
X : array-like of shape (n_samples_X, n_features)
16991707
A feature array.
17001708
1701-
Y : ndarray of shape (n_samples_Y, n_features), default=None
1709+
Y : array-like of shape (n_samples_Y, n_features), default=None
17021710
An optional second feature array. If `None`, uses `Y=X`.
17031711
17041712
gamma : float, default=1

sklearn/tests/test_public_functions.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -235,6 +235,7 @@ def _check_function_param_validation(
235235
"sklearn.metrics.pair_confusion_matrix",
236236
"sklearn.metrics.adjusted_rand_score",
237237
"sklearn.metrics.pairwise.additive_chi2_kernel",
238+
"sklearn.metrics.pairwise.chi2_kernel",
238239
"sklearn.metrics.pairwise.cosine_distances",
239240
"sklearn.metrics.pairwise.cosine_similarity",
240241
"sklearn.metrics.pairwise.euclidean_distances",

0 commit comments

Comments
 (0)