Code for F-GAM

This repository contains the PyTorch implementation for the paper A Factored Generalized Additive Model for Clinical Decision Support in the Operating Room.

Dependencies:

• Python 2 or 3
• PyTorch >= 0.4.0
• numpy >= 1.16.0
• scipy >= 1.2.0
• sklearn >= 0.18.0

Train a F-GAM using randomly generated data

python fgam.py

We replaced our ACTFAST data with randomly generated data in My_DataLoader.py

Questions?

Please send emails to me if you have any questions!