Commit 8be9afcc authored by schnurr's avatar schnurr
Browse files

Update sentence_similarity.py

parent ff4e7c11
Loading
Loading
Loading
Loading
+1 −0
Changes for src/sentence_similarity.py: 1 added line, 0 removed lines.
Original line number Diff line number Diff line
from src.SBERT_Model import SBERT_Model
from STSB_Dataset import STSB_Dataset
import probing

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