@@ -40,7 +40,8 @@ def trustworthiness(
4040) -> float : ...
4141
4242class 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