Commit a1e1fd8c authored by schaper's avatar schaper
Browse files

Update STSB_Dataset.py with _as_dataframe attribute

parent f60f6c58

STSB_Dataset.py

+4 −1
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):
@@ -33,6 +34,8 @@ class STSB_Dataset(lit_dataset.Dataset): 
          'label': label,

        } for dp, label in zip(self.data, self.labels)]



        self._as_dataframe = pd.DataFrame.from_dict(dataset._examples)



    def spec(self):

        return {

          'sentence1': lit_types.TextSegment(),