Commit 92f83aa9 authored by schaper's avatar schaper
Browse files

Update sentence_similarity.py

parent 6ed4931e

sentence_similarity.py

+1 −1
Original line number Diff line number Diff line 
from SBERT_Model import SBERT_Model

from STSB_Dataset import STSB_Dataset



dataset = STSB_Dataset('data/stsbenchmark/sts-test.csv')

dataset = STSB_Dataset('data/stsbenchmark/sts-test.csv')._as_dataframe



zero_model = SBERT_Model("Zero Model", 'models/nli-bert-large/', dataset)

ft_model = SBERT_Model("Fine-tuned Model", 'models/stsb-bert-large/', dataset)