Commit 95d1bbb4 authored by schnurr's avatar schnurr
Browse files

Update sentence_similarity.py

parent 8be9afcc

src/sentence_similarity.py

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

from STSB_Dataset import STSB_Dataset

import probing

# import probing # when needed



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