Skip to content

Few-shot classification in Named Entity Recognition Task

Notifications You must be signed in to change notification settings

Fritz449/ProtoNER

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ProtoNER

Few-shot classification in Named Entity Recognition Task


This is the code accompanying the paper "Few-shot classification in Named Entity Recognition Task" by Alexander Fritzler, Varvara Logacheva and Maksim Kretov.

Please make sure you read the paper before you try to use or analyse this code. This repository contains 4 folders that allow you to run a specific model using Allennlp (v0.6.0) framework.

  1. simple_base

This is a model that trains on a small dataset from scratch.

  1. warming

This model trains on big dataset that doesn't contain objects from the validation class.

  1. warm_base

This model loads from initialization generated by the previous warming model and trains as "simple_base" model.

  1. warm_protonet

This model loads from initialization generated by the warming model and trains by a procedure described in the paper

To launch the model run from the corresponding folder:

python global_experiment.py

If you have a question about this code, please contact me by [email protected]

About

Few-shot classification in Named Entity Recognition Task

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages