Update README.md

README.md

@@ -5,24 +5,29 @@ tags:
 - imbalanced-classification
 ---
 
-## Model description
-
-More information needed
+## Model Description
+### Keras Implementation of Imbalanced classification: credit card fraud detection
+This repo contains the trained model of [Imbalanced classification: credit card fraud detection](https://keras.io/examples/structured_data/imbalanced_classification/).
+The full credit goes to: [fchollet](https://twitter.com/fchollet)
 
 ## Intended uses & limitations
+- The trained model is used to detect of a specific transaction is fraudulent or not.
 
-More information needed
-
-## Training and evaluation data
-
-More information needed
+## Training dataset
+- [Credit Card Fraud Detection](https://www.kaggle.com/datasets/mlg-ulb/creditcardfraud)
+- Due to the high imbalance of the target feature (417 frauds or 0.18% of total 284,807 samples), training weight was applied to reduce the False Negatives to the lowest level as possible.
 
 ## Training procedure
-
-### Training hyperparameters
-
+### Training hyperparameter 
 The following hyperparameters were used during training:
-- optimizer: {'name': 'Adam', 'learning_rate': 0.01, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
+- optimizer: 'Adam'
+- learning_rate: 0.01
+- loss: 'binary_crossentropy'
+- epochs: 30
+- batch_size: 2048
+- beta_1: 0.9
+- beta_2: 0.999
+- epsilon: 1e-07
 - training_precision: float32
 
  ## Training Metrics
@@ -59,6 +64,7 @@ The following hyperparameters were used during training:
 | 28| 0.0|  9.0|  5305.0|  222124.0|  408.0|  0.071|  0.978|  0.026|  9.0|  398.0|  56488.0|  66.0|  0.142|  0.88| 
 | 29| 0.0|  5.0|  4846.0|  222583.0|  412.0|  0.078|  0.988|  0.242|  6.0|  7883.0|  49003.0|  69.0|  0.009|  0.92| 
 | 30| 0.0|  5.0|  5193.0|  222236.0|  412.0|  0.074|  0.988|  0.026|  7.0|  449.0|  56437.0|  68.0|  0.132|  0.907| 
+
  ## Model Plot
 
 <details>