Ejemplo n.º 1
0
    }


class FashionMNISTTest(MNISTTest):
    DATASET_CLASS = mnist.FashionMNIST


class KMNISTTest(MNISTTest):
    DATASET_CLASS = mnist.KMNIST


mnist.EMNIST.BUILDER_CONFIGS.extend([
    mnist.EMNISTConfig(
        name="test",
        class_number=200,
        train_examples=10,
        test_examples=2,
        description="EMNIST test data config.",
    ),
])


class EMNISTTest(testing.DatasetBuilderTestCase):
    DATASET_CLASS = mnist.EMNIST
    SPLITS = {
        "train": 10,
        "test": 2,
    }
    BUILDER_CONFIG_NAMES_TO_TEST = ["test"]

Ejemplo n.º 2
0
class KMNISTTest(MNISTTest):
    DATASET_CLASS = mnist.KMNIST


class KMNISTTestS3(KMNISTTest):
    VERSION = "2.0.0"


mnist.EMNIST.BUILDER_CONFIGS.extend([
    mnist.EMNISTConfig(
        name="test",
        class_number=200,
        train_examples=10,
        test_examples=2,
        description="EMNIST test data config.",
        version="1.0.1",
        supported_versions=[
            tfds.core.Version("2.0.0",
                              experiments={tfds.core.Experiment.S3: True}),
        ],
    ),
])


class EMNISTTest(testing.DatasetBuilderTestCase):
    DATASET_CLASS = mnist.EMNIST
    SPLITS = {
        "train": 10,
        "test": 2,
    }
    BUILDER_CONFIG_NAMES_TO_TEST = ["test"]