search

gregversteeg / corex_topic

Hierarchical unsupervised and semi-supervised topic models for sparse count data with CorEx
Apache License 2.0
626 stars 119 forks source link

GPU Implementation #41

Open RyanCodrai opened 4 years ago

RyanCodrai commented 4 years ago

This is an enhancement. Given that CorEx utilises a semi-supervised approach it would be advantageous to have a GPU Implementation as the reduced wait for feedback would allow for more rapid development of topics. https://developer.nvidia.com/how-to-cuda-python

gregversteeg commented 4 years ago

Hi Ryan, that's a good idea. There's not a GPU implementation for the topic models, but some variations have been coded in PyTorch: https://github.com/hrayrhar/T-CorEx