Skip to content

classical model code implementation of few-shot/one-shot lenaring, including siamese network, prototypical network, relation network, induction network

Notifications You must be signed in to change notification settings

RoryShao/few_shot_learning

Repository files navigation

few-shot learning

Each folder contains an implementation of each model.
  • data_helper. data processing
  • model. model construction
  • trainer. train model
  • metrics. performance metrics
  • config.json Configuration files for model parameters and training parameters

induction_network

  • paper: Few-Shot Text Classification with Induction Network

relation_network

  • paper: Learning to Compare: Relation Network for Few-Shot Learning

prototypical_network

  • paper: Prototypical Networks for Few-shot Learning

siamese_network

  • paper: Siamese Neural Networks for One-shot Image Recognition

data

  • the data from Amazon Review Data Set
  • citation: Image-based recommendations on styles and substitutes J. McAuley, C. Targett, J. Shi, A. van den Hengel SIGIR, 2015
  • link: you can download data from

note

  • You can only use 5-way, and if you need to use other way, you can modify the data_helper.py file.
  • Shot should not be more than 10, because there are few comments under some categories.
  • a smaller data set is provided here, you can get the larger data set from: 链接:https://pan.baidu.com/s/1b3JDjrRXgdOL0NKjX-QHbQ 提取码:2ieu

About

classical model code implementation of few-shot/one-shot lenaring, including siamese network, prototypical network, relation network, induction network

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%