Description

Code for the paper

Undirected Probabilistic Model for Tensor Decomposition

[paper]

@inproceedings{
tao2023undirected,
title={Undirected Probabilistic Model for Tensor Decomposition},
author={Zerui Tao and Toshihisa Tanaka and Qibin Zhao},
booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
year={2023},
url={https://openreview.net/forum?id=3NWWgB2SuF}
}

The code is basically builded on PyTorch. Detailed requirements are listed in ./requirements.txt.

We provide several examples.

1. Simulation for mixture of Gaussian distribution: ./simulation_mog.ipynb
2. Simulation for dynamic tensor completion: ./simulation_dynamic.ipynb
3. Experiment for Beijing Air dataset: ./exp_beijing_air.py