/MoNIG

Code for paper Trustworthy Multimodal Regression with Mixture of Normal-inverse Gamma Distributions.

Primary LanguagePython

MoNIG

Code for paper Trustworthy Multimodal Regression with Mixture of Normal-inverse Gamma Distributions.

Environment：

pytorch 1.0.2 matplotlib 3.3.4 numpy 1.19.2 python 3.7.10 scipy 1.6.1 vs2015_runtime 14.27.29016 xlrd 1.2.0

To train the model in the paper, run this example command:

python main.py --model MOE --dataset mosei_senti