├── .github └── workflows │ └── main.yml ├── .gitignore ├── README.md ├── data_processing └── gtex │ ├── .ipynb_checkpoints │ └── Untitled-checkpoint.ipynb │ ├── Untitled.ipynb │ └── process_gtex_data.py ├── dist ├── multigroupGP-0.4.tar.gz ├── multigroupGP-0.5.tar.gz ├── multigroupGP-0.6.tar.gz ├── multigroupGP-0.7.tar.gz ├── multigroupGP-0.9.tar.gz ├── multigroupGP-1.0.tar.gz ├── multigroupGP-1.1.tar.gz ├── multigroupGP-1.2.tar.gz ├── multigroupGP-1.3.tar.gz ├── multigroupGP-1.4.tar.gz └── multigroupGP-1.5.tar.gz ├── examples ├── gp_predictions.py ├── images │ ├── a_visualization.png │ ├── gp_preds.png │ └── mggp_preds.png ├── mggp_predictions.py └── visualize_a.py ├── experiments ├── gtex │ ├── gtex_LL_over_a_plot.py │ ├── gtex_find_pairwise_as.py │ ├── gtex_mggp_vs_hgp.py │ ├── gtex_pca_embedding.py │ ├── gtex_posterior_inference.py │ ├── gtex_posterior_inference_mll.py │ ├── gtex_prediction.py │ ├── gtex_single_genes.py │ ├── gtex_two_group.py │ ├── out │ │ └── stan_out │ │ │ └── cov_params_samples.csv │ ├── plot_a_heatmap.R │ ├── plot_ab_joint_density.py │ ├── process_gtex_data.py │ └── process_stan_output.py └── simulations │ ├── gp_stan.py │ ├── group_imbalance.py │ ├── mggp_frequentist_consistency.py │ ├── mggp_simple_viz.py │ ├── mll_comparison.py │ ├── posterior_figure.py │ ├── posterior_inference.py │ ├── posterior_inference_latent_process.py │ ├── posterior_inference_latent_process_multivariable.py │ ├── predictions.py │ ├── process_stan_output.py │ ├── recover_a.py │ └── recover_all_kernel_params.py ├── multigroupGP.egg-info └── PKG-INFO ├── multigroupGP ├── __init__.py ├── kernels │ ├── __init__.py │ ├── kernels.py │ └── util.py ├── models │ ├── __init__.py │ ├── gp-fit.stan │ ├── gp.py │ ├── gp_collapsed.stan │ ├── mggp_collapsed.stan │ ├── mggp_collapsed_fixed_ab.stan │ ├── mggp_multivariable.stan │ └── mggp_uncollapsed.stan └── old │ ├── benchmark_jax.py │ ├── gp_autograd.py │ ├── kernels_autograd.py │ └── recover_all_kernel_parameters.py ├── requirements.txt ├── setup.py └── tests ├── test_embeddings.py ├── test_gp_preds.py ├── test_jax_speed.py └── test_kernels.py /.github/workflows/main.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/.github/workflows/main.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/README.md -------------------------------------------------------------------------------- /data_processing/gtex/.ipynb_checkpoints/Untitled-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/data_processing/gtex/.ipynb_checkpoints/Untitled-checkpoint.ipynb -------------------------------------------------------------------------------- /data_processing/gtex/Untitled.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/data_processing/gtex/Untitled.ipynb -------------------------------------------------------------------------------- /data_processing/gtex/process_gtex_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/data_processing/gtex/process_gtex_data.py -------------------------------------------------------------------------------- /dist/multigroupGP-0.4.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/dist/multigroupGP-0.4.tar.gz -------------------------------------------------------------------------------- /dist/multigroupGP-0.5.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/dist/multigroupGP-0.5.tar.gz -------------------------------------------------------------------------------- /dist/multigroupGP-0.6.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/dist/multigroupGP-0.6.tar.gz -------------------------------------------------------------------------------- /dist/multigroupGP-0.7.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/dist/multigroupGP-0.7.tar.gz -------------------------------------------------------------------------------- /dist/multigroupGP-0.9.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/dist/multigroupGP-0.9.tar.gz -------------------------------------------------------------------------------- /dist/multigroupGP-1.0.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/dist/multigroupGP-1.0.tar.gz -------------------------------------------------------------------------------- /dist/multigroupGP-1.1.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/dist/multigroupGP-1.1.tar.gz -------------------------------------------------------------------------------- /dist/multigroupGP-1.2.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/dist/multigroupGP-1.2.tar.gz -------------------------------------------------------------------------------- /dist/multigroupGP-1.3.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/dist/multigroupGP-1.3.tar.gz -------------------------------------------------------------------------------- /dist/multigroupGP-1.4.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/dist/multigroupGP-1.4.tar.gz -------------------------------------------------------------------------------- /dist/multigroupGP-1.5.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/dist/multigroupGP-1.5.tar.gz -------------------------------------------------------------------------------- /examples/gp_predictions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/examples/gp_predictions.py -------------------------------------------------------------------------------- /examples/images/a_visualization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/examples/images/a_visualization.png -------------------------------------------------------------------------------- /examples/images/gp_preds.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/examples/images/gp_preds.png -------------------------------------------------------------------------------- /examples/images/mggp_preds.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/examples/images/mggp_preds.png -------------------------------------------------------------------------------- /examples/mggp_predictions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/examples/mggp_predictions.py -------------------------------------------------------------------------------- /examples/visualize_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/examples/visualize_a.py -------------------------------------------------------------------------------- /experiments/gtex/gtex_LL_over_a_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/gtex_LL_over_a_plot.py -------------------------------------------------------------------------------- /experiments/gtex/gtex_find_pairwise_as.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/gtex_find_pairwise_as.py -------------------------------------------------------------------------------- /experiments/gtex/gtex_mggp_vs_hgp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/gtex_mggp_vs_hgp.py -------------------------------------------------------------------------------- /experiments/gtex/gtex_pca_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/gtex_pca_embedding.py -------------------------------------------------------------------------------- /experiments/gtex/gtex_posterior_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/gtex_posterior_inference.py -------------------------------------------------------------------------------- /experiments/gtex/gtex_posterior_inference_mll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/gtex_posterior_inference_mll.py -------------------------------------------------------------------------------- /experiments/gtex/gtex_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/gtex_prediction.py -------------------------------------------------------------------------------- /experiments/gtex/gtex_single_genes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/gtex_single_genes.py -------------------------------------------------------------------------------- /experiments/gtex/gtex_two_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/gtex_two_group.py -------------------------------------------------------------------------------- /experiments/gtex/out/stan_out/cov_params_samples.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/out/stan_out/cov_params_samples.csv -------------------------------------------------------------------------------- /experiments/gtex/plot_a_heatmap.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/plot_a_heatmap.R -------------------------------------------------------------------------------- /experiments/gtex/plot_ab_joint_density.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/plot_ab_joint_density.py -------------------------------------------------------------------------------- /experiments/gtex/process_gtex_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/process_gtex_data.py -------------------------------------------------------------------------------- /experiments/gtex/process_stan_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/gtex/process_stan_output.py -------------------------------------------------------------------------------- /experiments/simulations/gp_stan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/gp_stan.py -------------------------------------------------------------------------------- /experiments/simulations/group_imbalance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/group_imbalance.py -------------------------------------------------------------------------------- /experiments/simulations/mggp_frequentist_consistency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/mggp_frequentist_consistency.py -------------------------------------------------------------------------------- /experiments/simulations/mggp_simple_viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/mggp_simple_viz.py -------------------------------------------------------------------------------- /experiments/simulations/mll_comparison.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/mll_comparison.py -------------------------------------------------------------------------------- /experiments/simulations/posterior_figure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/posterior_figure.py -------------------------------------------------------------------------------- /experiments/simulations/posterior_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/posterior_inference.py -------------------------------------------------------------------------------- /experiments/simulations/posterior_inference_latent_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/posterior_inference_latent_process.py -------------------------------------------------------------------------------- /experiments/simulations/posterior_inference_latent_process_multivariable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/posterior_inference_latent_process_multivariable.py -------------------------------------------------------------------------------- /experiments/simulations/predictions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/predictions.py -------------------------------------------------------------------------------- /experiments/simulations/process_stan_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/process_stan_output.py -------------------------------------------------------------------------------- /experiments/simulations/recover_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/recover_a.py -------------------------------------------------------------------------------- /experiments/simulations/recover_all_kernel_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/experiments/simulations/recover_all_kernel_params.py -------------------------------------------------------------------------------- /multigroupGP.egg-info/PKG-INFO: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP.egg-info/PKG-INFO -------------------------------------------------------------------------------- /multigroupGP/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/__init__.py -------------------------------------------------------------------------------- /multigroupGP/kernels/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multigroupGP/kernels/kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/kernels/kernels.py -------------------------------------------------------------------------------- /multigroupGP/kernels/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/kernels/util.py -------------------------------------------------------------------------------- /multigroupGP/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multigroupGP/models/gp-fit.stan: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/models/gp-fit.stan -------------------------------------------------------------------------------- /multigroupGP/models/gp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/models/gp.py -------------------------------------------------------------------------------- /multigroupGP/models/gp_collapsed.stan: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/models/gp_collapsed.stan -------------------------------------------------------------------------------- /multigroupGP/models/mggp_collapsed.stan: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/models/mggp_collapsed.stan -------------------------------------------------------------------------------- /multigroupGP/models/mggp_collapsed_fixed_ab.stan: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/models/mggp_collapsed_fixed_ab.stan -------------------------------------------------------------------------------- /multigroupGP/models/mggp_multivariable.stan: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/models/mggp_multivariable.stan -------------------------------------------------------------------------------- /multigroupGP/models/mggp_uncollapsed.stan: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/models/mggp_uncollapsed.stan -------------------------------------------------------------------------------- /multigroupGP/old/benchmark_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/old/benchmark_jax.py -------------------------------------------------------------------------------- /multigroupGP/old/gp_autograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/old/gp_autograd.py -------------------------------------------------------------------------------- /multigroupGP/old/kernels_autograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/old/kernels_autograd.py -------------------------------------------------------------------------------- /multigroupGP/old/recover_all_kernel_parameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/multigroupGP/old/recover_all_kernel_parameters.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/tests/test_embeddings.py -------------------------------------------------------------------------------- /tests/test_gp_preds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/tests/test_gp_preds.py -------------------------------------------------------------------------------- /tests/test_jax_speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/tests/test_jax_speed.py -------------------------------------------------------------------------------- /tests/test_kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andrewcharlesjones/multi-group-GP/HEAD/tests/test_kernels.py --------------------------------------------------------------------------------