search

plkmo / BERT-Relation-Extraction

PyTorch implementation for "Matching the Blanks: Distributional Similarity for Relation Learning" paper
Apache License 2.0
565 stars 132 forks source link

Fewrel task #5

Closed arunzz closed 4 years ago

arunzz commented 4 years ago

Hi

How do I use this for few-shot classification task. Pls advise.

plkmo commented 4 years ago

For fewrel task, to infer an input, you have to take its output representation vector (from the trained EM-MTB model) and do a dot product comparison with the exemplar's representations. I will package that in a script that in due time, since right now I'm a little busy and don't yet have a satisfactory trained EM-MTB model, which takes a long time to optimize with current resources.

vpvsankar commented 4 years ago

Did you able to get satisfactory result for MTB model? Also can you provide codes for FewRel task

plkmo commented 4 years ago

fewrel 5 way 1 shot task added. MTB training requires more GPU than i can handle currently.