TextClassificationBenchmark icon indicating copy to clipboard operation
TextClassificationBenchmark copied to clipboard

Add BERT benchmarks

Open loretoparisi opened this issue 7 years ago • 2 comments

Thanks for this repo. It could be very interesting to add the latest Google's BERT model that claims to be state-of-the-art in recent NLP tasks among them text classification. They have a classifier implementation to adapt, here some details.

loretoparisi avatar Nov 23 '18 14:11 loretoparisi

Actually, we are doting this, Please refer to https://github.com/wabyking/TextClassificationBenchmark/blob/master/models/BERTFast.py It depends on a related project named #pytorch_pretrained_bert#

wabyking avatar Nov 23 '18 15:11 wabyking

@wabyking thank you! I have just seen that 👍
By the way I was actually referring to BERT for Tensorflow rather than Pytorch as in the original implementation.

loretoparisi avatar Nov 23 '18 15:11 loretoparisi