-
-
Notifications
You must be signed in to change notification settings - Fork 25.8k
WIP Modifying nmf.py for accepting mini-batches #13386
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
Conversation
…ent) and benchmart file (WIP)
The advantage of the batch mode is that 'fit' calls 'fit_transform', so it automatically transforms the training data. This is not the case for the online method, so a 'transform' time should be considered |
@pcerda are you still working on this? |
I will try to pitch in. |
I'm currently working on something else. |
Working on it in #16948. |
I am closing this PR because #16948 is merged which includes the proposed feature. |
Reference Issues/PRs
#13308
What does this implement/fix? Explain your changes.
Modification of the NMF class to accept mini batches. For the moment, it works only with
solver='mu'
andbeta_loss='kullback-leibler
.I also added the benchmark code.
Any other comments?