/ocean

OCEAN: Online Task Inference for Compositional Tasks with Context Adaptation. [UAI 2020]

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

OCEAN

Implementation of OCEAN: Online Task Inference for Compositional Tasks with Context Adaptation.

Hongyu Ren, Yuke Zhu, Jure Leskovec, Anima Anandkumar and Animesh Garg. UAI 2020

Run

Check requirements.txt and examples.sh for example scripts of the experiments.

Cite

Please consider cite our paper if you find the paper and the code useful.

@inproceedings{ren2020ocean,
  title={OCEAN: Online Task Inference for Compositional Tasks with Context Adaptation},
  author={Ren, Hongyu and Zhu, Yuke and Leskovec, Jure and Anandkumar, Anima and Garg, Animesh}
  booktitle={Conference on Uncertainty in Artificial Intelligence},
  year={2020},
}

Feel free to send email to hyren@cs.stanford.edu if you have any questions. This code builds on PEARL.