Skip to content

Rename TSNE n_iter to max_iter#7500

Merged
rockhowse merged 2 commits intorapidsai:release/25.12from
jcrist:rename-tsne-n-iter
Nov 18, 2025
Merged

Rename TSNE n_iter to max_iter#7500
rockhowse merged 2 commits intorapidsai:release/25.12from
jcrist:rename-tsne-n-iter

Conversation

@jcrist
Copy link
Copy Markdown
Member

@jcrist jcrist commented Nov 17, 2025

In scikit-learn 1.5 they renamed the TSNE n_iter parameter to max_iter. In the interest of maximizing sklearn compatibility, here we do the same. This PR deprecates n_iter in favor of max_iter, in 26.02 we'll remove n_iter completely.

@jcrist jcrist self-assigned this Nov 17, 2025
@jcrist jcrist requested a review from a team as a code owner November 17, 2025 23:03
@jcrist jcrist requested a review from csadorf November 17, 2025 23:03
@jcrist jcrist added improvement Improvement / enhancement to an existing function breaking Breaking change labels Nov 17, 2025
@github-actions github-actions Bot added the Cython / Python Cython or Python issue label Nov 17, 2025
In scikit-learn 1.5 they renamed the TSNE `n_iter` parameter to
`max_iter`. In the interest of maximizing sklearn compatibility, here we
do the same. This PR deprecates `n_iter` in favor of `max_iter`, in
26.02 we'll remove `n_iter` completely.
@jcrist jcrist force-pushed the rename-tsne-n-iter branch from fc82e36 to d32aba2 Compare November 18, 2025 05:32
Copy link
Copy Markdown
Member

@betatim betatim left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

The formatting change to __init__ is on purpose or due to us changing formatting tool? Just to know if we should expect to see more of this in the future. Makes it minimally harder to figure out if the only thing that effectively changed is the value of n_iter

Comment thread python/cuml/cuml/manifold/t_sne.pyx Outdated
@betatim
Copy link
Copy Markdown
Member

betatim commented Nov 18, 2025

Restarted the failed build job. Looked like a networking error fetching the nccl package from anaconda.org

@rockhowse rockhowse merged commit 78cf0e7 into rapidsai:release/25.12 Nov 18, 2025
189 of 192 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

breaking Breaking change Cython / Python Cython or Python issue improvement Improvement / enhancement to an existing function

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants