| --- |
| library_name: tf-keras |
| tags: |
| - structured-data |
| - tabular-data |
| - classification |
| --- |
| # Tensorflow Keras Implementation of Structured data classification from scratch |
| This repo contains models and notebook for [Structured data classification from scratch](https://keras.io/examples/structured_data/structured_data_classification_from_scratch/). |
| This example demonstrates how to do structured data classification, starting from a raw CSV file. Our data includes both numerical and categorical features. We will use Keras preprocessing layers to normalize the numerical features and vectorize the categorical ones. |
| Full credits to [François Chollet](https://twitter.com/fchollet), creator of Keras! |
|
|
| ## Model description |
|
|
| The model is a very simple MLP with only one hidden layer. |
| This example showcases how to perform preprocessing of common tabular data *inside* a Keras model. |
| It uses tensorflow.keras.layers.{IntegerLookup, Normalization, StringLookup} to process numerical and categorical (integer or string) features. |
|
|
|
|
|
|
| ## Intended uses & limitations |
|
|
| This tool does not provide medical advice It is intended for informational purposes only. It is not a substitute for professional medical advice, diagnosis or treatment. |
|
|
| ## Training and evaluation data |
|
|
| [Our dataset](https://archive.ics.uci.edu/ml/datasets/heart+Disease) is provided by the Cleveland Clinic Foundation for Heart Disease. It's a CSV file with 303 rows. Each row contains information about a patient (a sample), and each column describes an attribute of the patient (a feature). We use the features to predict whether a patient has a heart disease (binary classification). |
| The model is trained on 80% of data and evaluated on remaining 20%. |
|
|
| ## Training procedure |
| Training proceeds for 50 epochs with default Adam optimizer on binary crossentropy. |
|
|
| ### Training hyperparameters |
|
|
| The following hyperparameters were used during training: |
| - optimizer: {'name': 'Adam', 'learning_rate': 0.001, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False} |
| - training_precision: float32 |
|
|
| ## Training Metrics |
| Training achieves >83% accuracy on the held-out validation data. |
| Check TensorBoard "Metrics" tab above for details. |
| ## Model Plot |
|
|
| <details> |
| <summary>View Model Plot</summary> |
|
|
|  |
|
|
| </details> |
