metadata
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.
Full credits go to Mehdi
Reproduced by Rushi Chaudhari
Siamese Networks 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 of handwritten digits
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
epochs = 10
batch_size = 16
margin = 1
Training results
View Model Plot
