项目作者: thbuerg

项目描述 :
🫀 Code for "Neural network-based integration of polygenic and clinical information: Development and validation of a prediction model for 10 year risk of major adverse cardiac events in the UK Biobank cohort" 🫀
高级语言: Jupyter Notebook
项目地址: git://github.com/thbuerg/NeuralCVD.git
创建时间: 2021-07-07T09:40:43Z
项目社区:https://github.com/thbuerg/NeuralCVD

开源协议:GNU General Public License v3.0

下载




Logo

Neural network-based integration of polygenic and clinical information: Development and validation of a prediction model for 10 year risk of major adverse cardiac events in the UK Biobank cohort

Paper00249-1/fulltext)

Description

Code related to the paper “Neural network-based integration of polygenic and clinical information: Development and validation of a prediction model for 10 year risk of major adverse cardiac events in the UK Biobank cohort”.
This repo is a python package for preprocessing UK Biobank data and preprocessing, training and evaluating the NeuralCVD score.

NeuralCVD

Methods

NeuralCVD is based on the fantastic Deep Survival Machines Paper, the original implementation can be found here.

Assets

This repo contains code to preprocess UK Biobank data, train the NeuralCVD score and analyze/evaluate its performance.

  • Preprocessing involves: parsing primary care records for desired diagnosis, aggregating the cardiovascular risk factors analyzed in the study and calculating predefined polygenic risk scores.
  • Training involves Model specification via pytorch-lightning and hydra.
  • Postprocessing involve extensive benchmarks with linear Models, and calculation of bootstrapped metrics.

How to train the NeuralCVD Model

  1. First, install dependencies
    ```bash

    clone project

    git clone https://github.com/thbuerg/NeuralCVD

install project

cd NeuralCVD
pip install -e .
pip install -r requirements.txt

  1. 2. Download UK Biobank data. Execute preprocessing notebooks on the downloaded data.
  2. 3. Edit the `.yaml` config files in `neuralcvd/experiments/config/`:
  3. ```yaml
  4. setup:
  5. project_name: <YourNeptuneSpace>/<YourProject>
  6. root_dir: absolute/path/to/this/repo/
  7. experiment:
  8. tabular_filepath: path/to/processed/data
  1. Set up Neptune.ai

  2. Train the NeuralCVD Model (make sure you are on a machine w/ GPU)
    ```bash

    module folder

    cd neuralcvd

run training

bash experiments/run_NeuralCVD_S.sh

  1. ## Citation

@article{steinfeldt2022neural,
title={Neural network-based integration of polygenic and clinical information: development and validation of a prediction model for 10-year risk of major adverse cardiac events in the UK Biobank cohort},
author={Steinfeldt, Jakob and Buergel, Thore and Loock, Lukas and Kittner, Paul and Ruyoga, Greg and zu Belzen, Julius Upmeier and Sasse, Simon and Strangalies, Henrik and Christmann, Lara and Hollmann, Noah and others},
journal={The Lancet Digital Health},
volume={4},
number={2},
pages={e84—e94},
year={2022},
publisher={Elsevier}
}
```