-
-
Notifications
You must be signed in to change notification settings - Fork 25.9k
ENH TfidfTransformer perserves np.float32 dtype #28136
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
ENH TfidfTransformer perserves np.float32 dtype #28136
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the PR @glemaitre. Should we have some tests to check the preservation?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. Thanks @glemaitre
Towards #11000
Follow-up to #18843
This PR allows the
TfidfTransformer
to make the computation usingnp.float32
without additional conversion.As a free launch, the
TfidfVectorizer
should make less copy when settingdtype=np.float32
.