A Pytorch Implementation of Attentive Neural Process
preprocess.py
includes all preprocessing codes when you loads data.module.py
contains all methods, including attention, linear and so on.network.py
contains whole structure of network.train.py
is for training ANP model.generate.ipynb
is for generating samples.test samples after 50 epoch training with random context selection.