Exemplo n.º 1
0
    def get_dataset(self, segment='train'):
        """Get the corresponding dataset for QQP.

        Parameters
        ----------
        segment : str, default 'train'
            Dataset segments. Options are 'train', 'dev', 'test'.
        """
        return GlueQQP(segment=segment)
Exemplo n.º 2
0
    def get_dataset(self, segment='train',
                    root=os.path.join(os.getenv('GLUE_DIR', 'glue_data'), 'QQP')):
        """Get the corresponding dataset for QQP.

        Parameters
        ----------
        segment : str, default 'train'
            Dataset segments. Options are 'train', 'dev', 'test'.
        root : str, default $GLUE_DIR/QQP
            Path to the folder which stores the dataset.
        """
        return GlueQQP(segment=segment, root=root)