princeton-nlp / LM-BFF

[ACL 2021] LM-BFF: Better Few-shot Fine-tuning of Language Models https://arxiv.org/abs/2012.15723

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Bug in tools/ensemble.py/get_labels

dyukha opened this issue · comments

The function returns label_ids. However, when print_name in ['sst-5', 'mr', 'cr', 'mpqa', 'subj', 'trec'], label_ids is not defined. I guess there should be return labels in this branch?

Thanks for reporting this! The bug has been fixed.