How to implement supervised LDA model more efficiently on PyMC