zshwuhan / Neural-Dynamic-Focused-Topic-Model

Neural Dynamic Focused Topic Model

Code accompanying the paper Neural Dynamic Focused Topic Model.

Installation of the library

In order to set up the necessary environment:

1. create an environment drf with the help of [conda],

   conda env create -f environment.yaml
   

2. activate the new environment with

   conda activate drf
   

3. install drf with:

   python setup.py install # or `develop`
   python setup.py install # or
   
   

Data Preprocessing and preparation

All the data used should be in the folder {project_location}/data. The preprocessing of the *NIPS, ACL and UN datasets is done using the script located in {project_location}/scripts/preprocessing/dataset_preprocessor.py.

Training of models

The training of the models used in the paper is done by using the training scripts located in the {project_location}/scripts/train/{acl|nips|un|nips_perrone} folder.

Evaluating of the trained models

For evaluation of the trained models we use the script {project_location}/scripts/evaluate/topic_models_evaluate_all.py