/SparseTensorHGP

Sparse Tensor Factorization with Hierarchical Gamma Processes

Primary LanguagePythonMIT LicenseMIT

SparseTensorHGP

Sparse Tensor Factorization with Hierarchical Gamma Processes

This is the implementation of the sparse tensor decomposition algorithms from the paper Nonparametric Sparse Tensor Factorization with Hierarchical Gamma Processes, by Conor Tillinghast, Zheng Wang and Shandian Zhe @ The Thirty-nineth International Conference on Machine Learning (ICML), 2022.

Citation

If you use our code please cite our paper

@article{ctill2022sparsehdp,
	title={Nonparametric Sparse Tensor Factorization with Hierarchical Gamma Processes}, 
	author={Tillinghast, Conor and Wang, Zheng and Zhe, Shandian},
	journal={Proceedings of the 39th International Conference on Machine Learning},
	year={2022}}