Global Transformed Prototypical Networks for Few-Shot Relation Classification

Codes for our paper "From Learning-to-Match to Learning-to-Discriminate: Global Prototype Learning for Few-shot Relation Classification"[paper]

Data preparation

we use the official FewRel dataset for training and evaluation, please download from https://github.com/thunlp/FewRel and put data files into ./data directory.

Usage

prerequirement

pip install -r requirements.txt

Train

python train_demo.py --mode train

Test

bash test.sh # replace the file path in this file