/pytorch_bert_similarity

Pytorch version of bert to get the similarities of two sentences.

Primary LanguagePython

pytorch_bert_similarity

Pytorch version of bert to get the similarities of two sentences.

Requirements:

python==3.6

pytorch==1.3.0

transformers==2.2.1

skleaern==0.20.4