Loading STSB_Dataset.py +2 −13 Original line number Diff line number Diff line from lit_nlp.api import dataset as lit_dataset from lit_nlp.api import types as lit_types import pandas as pd class STSB_Dataset(lit_dataset.Dataset): # this is a LIT-specific wrapper for our STS-B Dataset. class STSB_Dataset(): def __init__(self, path): Loading Loading @@ -35,10 +31,3 @@ class STSB_Dataset(lit_dataset.Dataset): } for dp, label in zip(self.data, self.labels)] self.as_dataframe = pd.DataFrame.from_dict(self._examples) No newline at end of file def spec(self): return { 'sentence1': lit_types.TextSegment(), 'sentence2': lit_types.TextSegment(), 'label': lit_types.RegressionScore(), } Loading
STSB_Dataset.py +2 −13 Original line number Diff line number Diff line from lit_nlp.api import dataset as lit_dataset from lit_nlp.api import types as lit_types import pandas as pd class STSB_Dataset(lit_dataset.Dataset): # this is a LIT-specific wrapper for our STS-B Dataset. class STSB_Dataset(): def __init__(self, path): Loading Loading @@ -35,10 +31,3 @@ class STSB_Dataset(lit_dataset.Dataset): } for dp, label in zip(self.data, self.labels)] self.as_dataframe = pd.DataFrame.from_dict(self._examples) No newline at end of file def spec(self): return { 'sentence1': lit_types.TextSegment(), 'sentence2': lit_types.TextSegment(), 'label': lit_types.RegressionScore(), }