buio commited on
Commit
55b09db
1 Parent(s): 29d90b7

Add model card

Browse files
Files changed (1) hide show
  1. README.md +12 -3
README.md CHANGED
@@ -5,20 +5,29 @@ tags:
5
  - tabular-data
6
  - classification
7
  ---
 
 
 
 
8
 
9
  ## Model description
10
 
11
- More information needed
 
 
 
 
12
 
13
  ## Intended uses & limitations
14
 
15
- More information needed
16
 
17
  ## Training and evaluation data
18
 
19
- More information needed
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