Code accompanying the paper: "Neural dynamic focused topic model." Proceedings of the AAAI Conference on Artificial Intelligence. Vol. 37. No. 11. 2023.
In order to set up the necessary environment:
create an environment drf
with the help of [conda],
conda env create -f environment.yaml
activate the new environment with
conda activate drf
install drf
with:
python setup.py install # or `develop`
python setup.py install # or
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
.
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.
For evaluation of the trained models we use the script {project_location}/scripts/evaluate/topic_models_evaluate_all.py