File size: 1,288 Bytes
ca172f6 d426fc1 ca172f6 cec0650 21f9c36 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
---
library_name: tf-keras
---
## Model description
This repo contains the model for the notebook [Image similarity estimation using a Siamese Network with a contrastive loss](https://keras.io/examples/vision/siamese_contrastive/).
Full credits go to Mehdi
Reproduced by [Rushi Chaudhari](https://github.com/rushic24)
[Siamese Networks](https://en.wikipedia.org/wiki/Siamese_neural_network) are neural networks which share weights between two or more sister networks, each producing embedding vectors of its respective inputs.
In supervised similarity learning, the networks are then trained to maximize the contrast (distance) between embeddings of inputs of different classes, while minimizing the distance between embeddings of similar classes, resulting in embedding spaces that reflect the class segmentation of the training inputs.
## Dataset
[MNIST dataset](https://www.tensorflow.org/datasets/catalog/mnist) of handwritten digits
## Training procedure
### Training hyperparameters
The following hyperparameters were used during training:
```
epochs = 10
batch_size = 16
margin = 1
```
### Training results
<details>
<summary> View Model Plot </summary>
</details> |