search

h2oai / h2o4gpu

H2Oai GPU Edition
Apache License 2.0
460 stars 95 forks source link

Add Latent Dirichlet Allocation to roadmap [feature request] #675

Open gregwchase opened 6 years ago

gregwchase commented 6 years ago

Description

After using the Gensim and PySpark LDA implementations, this is definitely an algorithm which could offer massive gains if developed for the GPU.

For reference, the SciKit-Learn implementation is linked below. http://scikit-learn.org/stable/modules/generated/sklearn.decomposition.LatentDirichletAllocation.html