Add model card
Browse files
README.md
CHANGED
|
@@ -5,20 +5,29 @@ tags:
|
|
| 5 |
- tabular-data
|
| 6 |
- classification
|
| 7 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
## Model description
|
| 10 |
|
| 11 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
## Intended uses & limitations
|
| 14 |
|
| 15 |
-
|
| 16 |
|
| 17 |
## Training and evaluation data
|
| 18 |
|
| 19 |
-
|
| 20 |
|
| 21 |
## Training procedure
|
|
|
|
| 22 |
|
| 23 |
### Training hyperparameters
|
| 24 |
|
|
|
|
| 5 |
- tabular-data
|
| 6 |
- classification
|
| 7 |
---
|
| 8 |
+
# Tensorflow Keras Implementation of Structured data classification from scratch
|
| 9 |
+
This repo contains models and notebook for [Structured data classification from scratch](https://keras.io/examples/structured_data/structured_data_classification_from_scratch/).
|
| 10 |
+
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.
|
| 11 |
+
Full credits to [François Chollet](https://twitter.com/fchollet), creator of Keras!
|
| 12 |
|
| 13 |
## Model description
|
| 14 |
|
| 15 |
+
The model is a very simple MLP with only one hidden layer.
|
| 16 |
+
This example showcases how to perform preprocessing of common tabular data *inside* a Keras model.
|
| 17 |
+
It uses tensorflow.keras.layers.{IntegerLookup, Normalization, StringLookup} to process numerical and categorical (integer or string) features.
|
| 18 |
+
|
| 19 |
+
|
| 20 |
|
| 21 |
## Intended uses & limitations
|
| 22 |
|
| 23 |
+
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.
|
| 24 |
|
| 25 |
## Training and evaluation data
|
| 26 |
|
| 27 |
+
[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).
|
| 28 |
|
| 29 |
## Training procedure
|
| 30 |
+
Training proceeds for 50 epochs with default Adam optimizer on binary crossentropy.
|
| 31 |
|
| 32 |
### Training hyperparameters
|
| 33 |
|