|
1 | 1 | # MOMoGP
|
2 |
| -Code for "Leveraging Probabilistic Circuits for Nonparametric Multi-Output Regression" @UAI2021 |
| 2 | + |
| 3 | +This is the official repository for MOMoGP introduced in |
| 4 | +[Leveraging Probabilistic Circuits for Nonparametric Multi-Output Regression](https://ml-research.github.io/papers/yu2021uai_momogps.pdf) by Zhongjie Yu, Mingye Zhu, Martin Trapp, Arseny Skryagin, and Kristian Kersting, to be published at UAI 2021. |
| 5 | + |
| 6 | + |
| 7 | + |
| 8 | + |
| 9 | +## Setup |
| 10 | + |
| 11 | +This will clone the repo, install a Python virtual env (requires Python 3.6), the required packages, and will download some datasets. |
| 12 | + |
| 13 | + git clone https://github.com/minimrbanana/MOMoGP |
| 14 | + ./setup.sh |
| 15 | + |
| 16 | +## Demos |
| 17 | + |
| 18 | +To illustrate the usage of the code: |
| 19 | + |
| 20 | + source ./venv_momogp/bin/activate |
| 21 | + python run_MOMoGP.py --data=parkinsons |
| 22 | + |
| 23 | +"parkinsons" can be replaced with "scm20d" or "wind" or "energy" or "usflight". |
| 24 | + |
| 25 | +### Hyperparameters |
| 26 | + |
| 27 | +If not specified, the corresponding hyperparameters are set by default values. |
| 28 | +If train on CPU, use: |
| 29 | + |
| 30 | + python run_MOMoGP.py --data=parkinsons --cpu |
| 31 | + |
| 32 | +## Citation |
| 33 | +If you find this code useful in your research, please consider citing: |
| 34 | + |
| 35 | + |
| 36 | + @inproceedings{yu2021uai_momogps, |
| 37 | + title = {Leveraging Probabilistic Circuits for Nonparametric Multi-Output Regression}, |
| 38 | + author = {Yu, Zhongjie and Zhu, Mingye and Trapp, Martin and Skryagin, Arseny and Kersting, Kristian}, |
| 39 | + booktitle = {Proceedings of the 37th Conference on Uncertainty in Artificial Intelligence (UAI)}, |
| 40 | + year = {2021} |
| 41 | + } |
| 42 | + |
| 43 | +## Acknowledgments |
| 44 | + |
| 45 | +* This work is supported by the Federal Ministry of Education and Research (BMBF; project "MADESI", FKZ 01IS18043B, and Competence Center for AI and Labour; "kompAKI", FKZ 02L19C150), the Hessian Ministry of Higher Education, Research, Science and the Arts (HMWK; projects "The Third Wave of AI" and "The Adaptive Mind"), the Hessian research priority programme LOEWE within the project "WhiteBox", and the National Research Center for Applied Cybersecurity ATHENE, a joint effort of BMBF and HMWK. M.T. acknowledges funding from the Academy of Finland (grant number 324345). |
| 46 | + |
| 47 | +* The code is developed based on the Python implementation of DSMGP from [Eugene](https://github.com/eugene/spngp). |
| 48 | + |
| 49 | + |
| 50 | + |
0 commit comments