Skip to content

Commit 9bd1ff0

Browse files
committed
uai 2021 v0
1 parent 76deaf3 commit 9bd1ff0

22 files changed

+1499186
-1
lines changed

MOMoGP.py

+420
Large diffs are not rendered by default.

MOMoGPstructure.py

+397
Large diffs are not rendered by default.

README.md

+49-1
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,50 @@
11
# 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+
![Learn_MOMoGP](./figures/Learn_MOMoGP.png)
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+

data/Parkinsons/x_test.csv

+1,763
Large diffs are not rendered by default.

data/Parkinsons/x_train.csv

+4,112
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)