Python EmnistDataset.load_or_generate_data Examples

Example #1

 def test_evaluate(self):
     predictor = CharacterPredictor()
     dataset = EmnistDataset()
     dataset.load_or_generate_data()
     t = time()
     metric = predictor.evaluate(dataset)
     time_taken = time() - t
     print(f'acc: {metric}, time_taken: {time_taken}')
     self.assertGreater(metric, 0.7)
     self.assertLess(time_taken, 10)

Example #2

File: create_emnist_support_files.py Project: ngbrenda/fsdl-text-recognizer-project-2018

def create_emnist_support_files():
    shutil.rmtree(SUPPORT_DIRNAME, ignore_errors=True)
    SUPPORT_DIRNAME.mkdir()

    dataset = EmnistDataset()
    dataset.load_or_generate_data()

    for ind in [5, 7, 9]:
        image = dataset.x_test[ind]
        label = dataset.mapping[np.argmax(dataset.y_test[ind])]
        print(ind, label)
        util.write_image(image, str(SUPPORT_DIRNAME / f'{label}.png'))