← all repositories

jakesnell/prototypical-networks

PyTorch implementation of prototypical networks for few-shot learning, a metric-learning approach to classification from few examples.

1.2k stars Python ML Frameworks
prototypical-networks
Velocity · 7d
+0.4
★ / day
Trend
steady
star history

This repository provides code for the prototypical networks paper (NeurIPS 2017), implementing an embedding-based few-shot classification method. The approach learns a metric space where class prototypes are computed from support examples, enabling classification from few labeled samples. It includes training and evaluation scripts with support for GPU acceleration and train/validation workflows.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.