Google Brain - Ventilator Pressure Prediction: Winner's Solution

This repo contains the PID matching code, a fundamental part of the winning solution for the Google Brain - Ventilator Pressure Prediction competition. A write-up can be found here.

The code of our two models can be found here:

1. LSTM on raw data
2. LSTM + CNN Transformer on features which is available on Kaggle as well.

The matching code is available in find_pressure_w_triangle.py but is rather messy. For a better explanation of the concepts, please check out our notebook on Kaggle. The Create Script notebook can be used to create bash scripts that will do the matching on different CPU's of your computer.

We also published multiple notebooks on Kaggle:

1. PID Matching
2. Train 1 Fold of the LSTM + CNN Transformer

And a dataset of pretrained models for the LSTM + CNN transformer model is available on Kaggle as well!