Ejemplos de Dataset.load_dataset en Python

Ejemplo n.º 1

Archivo: test_dataset.py Proyecto: satrusskumar/robustness-gym

    def test_load_dataset(self):
        # Load the first 20 examples of the boolq dataset
        dataset = Dataset.load_dataset("boolq", split="train[:20]")

        # Check that we got 20 examples
        self.assertTrue(isinstance(dataset, Dataset))
        self.assertEqual(len(dataset), 20)

Ejemplo n.º 2

    def test_load_dataset(self):
        # Load the first 20 examples of the boolq dataset
        dataset = Dataset.load_dataset("boolq",
                                       split="train[:20]",
                                       dataset_fmt="in_memory")

        # Check that we got 20 examples
        self.assertTrue(isinstance(dataset, Dataset))
        self.assertEqual(len(dataset), 20)
        self.assertTrue(
            isinstance(dataset.identifier.parameters["version"], str))