Skip to content

Commit 3be0222

Browse files
authored
Add TSNE param max_iter and deprecate n_iter in scikit-learn (#354)
1 parent b790aad commit 3be0222

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

stubs/sklearn/manifold/_t_sne.pyi

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,8 @@ def trustworthiness(
4040
) -> float: ...
4141

4242
class TSNE(BaseEstimator):
43-
n_iter_: int = ...
43+
max_iter: None | int = ...
44+
n_iter_: str | int = ...
4445
learning_rate_: float = ...
4546
feature_names_in_: ndarray = ...
4647
n_features_in_: int = ...
@@ -62,7 +63,7 @@ class TSNE(BaseEstimator):
6263
perplexity: Float = 30.0,
6364
early_exaggeration: Float = 12.0,
6465
learning_rate: float | Literal["auto"] = "auto",
65-
n_iter: Int = 1000,
66+
max_iter: None | Int = None,
6667
n_iter_without_progress: Int = 300,
6768
min_grad_norm: Float = 1e-7,
6869
metric: str | Callable = "euclidean",
@@ -74,6 +75,7 @@ class TSNE(BaseEstimator):
7475
angle: Float = 0.5,
7576
n_jobs: None | Int = None,
7677
square_distances: str | bool = "deprecated",
78+
n_iter: str | Int = "deprecated",
7779
) -> None: ...
7880
def fit_transform(self, X: MatrixLike, y: None | ndarray = None) -> ndarray: ...
7981
def fit(self, X: MatrixLike, y=None) -> ndarray: ...

0 commit comments

Comments
 (0)