Folders and files

Name		Name	Last commit message	Last commit date
configs		configs
data		data
dataset		dataset
model		model
train		train
.gitignore		.gitignore
README.md		README.md
argument.py		argument.py
main.py		main.py
requirments.txt		requirments.txt
test.py		test.py

README

This repository contains a pytorch implementation of Multi-Task Neural Processes.

Basic Usage

Training with missing rate 0.5

python main.py --model [mtp/stp/jtp/mtp_s] --data [synthetic/weather] --gamma_train 0.5

Testing with missing rate 0.5 and context size 10

python test.py --eval_name [mtp/stp/jtp/mtp_s] --data [synthetic/weather] --gamma 0.5 --cs 10

Citation

If you find this work useful, please consider citing:

@inproceedings{kim2021multi,
  title={Multi-Task Processes},
  author={Kim, Donggyun and Cho, Seongwoong and Lee, Wonkwang and Hong, Seunghoon},
  booktitle={International Conference on Learning Representations},
  year={2021}
}