/hddm

Primary LanguagePython

Hierarchical Deep Document Model

To run,

python main.py --batch_size 64 --dir_logs "./out/wiki_company/" --num_epochs 100 --num_topics "128 64 32" --layer_sizes "512 256 128" --embedding_sizes "100 50 25" --dataset wiki_company

Citation

@article{hddm,
    title={Hierarchical Deep Document Model},
    author={Yang, Yi and Lalor, John P. and Abbasi, Ahmed  and Zeng, Daniel Dajun },
    year={2024},
    journal={IEEE Transactions on Knowledge and Data Engineering},
    }