Commit 21278ee7 authored by Daniel Podrazka's avatar Daniel Podrazka
Browse files

Fix a typo in sentence_similarity.py

parent 6523f8a0
Loading
Loading
Loading
Loading
+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')._as_dataframe
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)