Commit 6523f8a0 authored by Daniel Podrazka's avatar Daniel Podrazka
Browse files

Remove the LIT-specific things in the STSB_Dataset class

parent 4e988dcf
Loading
Loading
Loading
Loading
+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):

@@ -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(),
        }