NMF_trainer
===========

.. py:module:: topmost.trainers.basic.NMF_trainer



Module Contents
---------------



.. autoapisummary::


   topmost.trainers.basic.NMF_trainer.NMFGensimTrainer


   topmost.trainers.basic.NMF_trainer.NMFSklearnTrainer





.. py:class:: NMFGensimTrainer(dataset, num_topics=50, num_top_words=15, max_iter=1)

   .. py:attribute:: dataset


   .. py:attribute:: num_topics
      :value: 50



   .. py:attribute:: num_top_words
      :value: 15



   .. py:attribute:: vocab_size


   .. py:attribute:: max_iter
      :value: 1



   .. py:method:: train()


   .. py:method:: test(bow)


   .. py:method:: get_beta()


   .. py:method:: get_top_words(num_top_words=None)


   .. py:method:: export_theta()


.. py:class:: NMFSklearnTrainer(model, dataset, num_top_words=15)

   .. py:attribute:: model


   .. py:attribute:: dataset


   .. py:attribute:: num_top_words
      :value: 15



   .. py:method:: train()


   .. py:method:: test(bow)


   .. py:method:: get_beta()


   .. py:method:: get_top_words(num_top_words=None)


   .. py:method:: export_theta()