Update README.md

README.md

@@ -1,34 +1,32 @@
 ---
-library_name: keras
+license: gpl-3.0
 ---
 
-## Model description
+## Model Description
 
-More information needed
+### Keras Implementation of Classification using Attention-based Deep Multiple Instance Learning (MIL)
 
-## Intended uses & limitations
+This repo contains the trained model of Classification using Attention-based Deep Multiple Instance Learning (MIL).
 
-More information needed
+Spaces Link:
+Keras Example Link:- https://keras.io/examples/vision/attention_mil_classification/
+## Intended uses & limitations
+- The trained model can be used to classify a bag of image instances (the bag of image instances can be generated from an original image) with the motivation of knowing which patterns in the original image is actually causing it to belong to that class.
 
 ## Training and evaluation data
-
-More information needed
+- Original MNIST train & test dataset were loaded from tensorflow datasets and the images were randomly chosen to create different bags of instance with number of instances per bag is 3
+- Number 8 is selected to be the Positive class (i.e. the bag that contains image of 8 belongs to Positive class and others are Negative class)
 
 ## Training procedure
-
-### Training hyperparameters
-
+### Training hyperparameter 
 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
-Model history needed
- ## Model Plot
+- optimizer: 'adam'
+- loss: 'sparse_categorical_crossentropy'
+- epochs: 50
+- batch_size: 8
 
+## Model Plot
 <details>
 <summary>View Model Plot</summary>
-
 ![Model Image](./model.png)
-
 </details>