Loading 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')._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) Loading Loading
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')._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) Loading