zichuan-liu / TimeXplusplus

[ICML'24] Official PyTorch Implementation of TimeX++
https://arxiv.org/pdf/2405.09308
20 stars 4 forks source link

tint module #2

Closed FlorentF9 closed 6 months ago

FlorentF9 commented 6 months ago

Hi, thanks for releasing your code!

I wondered what is the module "tint" used in bc_model.py ?

from tint.models import MLP, RNN

It does not seem to be in the requirements and I could not find the corresponding package (there is a package tint but it is about colors, not MLPs or RNNs..)

zichuan-liu commented 6 months ago

Hi, thank you for your attention, The package tint is from time_interpret and it should be installed as pip install time_interpret

zichuan-liu commented 6 months ago

I added this package to the requirements file, where MLPs in tint are just more convenient for constructing reparameterized distributions in DistributionParams for $KL(PX , P{tilde_X})$.

Thanks again.