This notebook demonstrates how to use the attributions explainer API to explain a Cleveland Heart Disease dataset classification example using a Custom TensorFlow NN.
heart_disease.ipynb
performs the following steps:
- Import dependencies
- Load and process the Heart Disease CSV file from the TensorFlow data hub
- Design the TensorFlow NN
- Train the CNN
- Visualize the custom NN classifications using
kernel_explainer()
To run heart_disease.ipynb
, install the following dependencies:
Heart Disease Dataset - https://archive.ics.uci.edu/ml/datasets/heart+disease