— PyTorch implementations of the ProtoNet variations
git clone https://github.com/borhanMorphy/prototypical-networks.git
cd prototypical-networks
pip install .
| Name | Extra Parameters | Training Metric | Inference Metric |
|---|---|---|---|
| ProtoNet | No | squared euclidean | squared euclidean |
| SEN-ProtoNet | No | sen | cosine |
| TapNet | Yes | squared euclidean | squared euclidean |
@inproceedings{snell2017prototypical,
title={Prototypical Networks for Few-shot Learning},
author={Snell, Jake and Swersky, Kevin and Zemel, Richard},
booktitle={Advances in Neural Information Processing Systems},
year={2017}
}