diff --git a/skluc/test/test_data/test_transformation/TestVGG19Transformer.py b/skluc/test/test_data/test_transformation/TestVGG19Transformer.py index ce3c33a20744165125263665399e3bef43586819..d5aca716ae11ec9bf54ddb405028e59414e12f15 100644 --- a/skluc/test/test_data/test_transformation/TestVGG19Transformer.py +++ b/skluc/test/test_data/test_transformation/TestVGG19Transformer.py @@ -1,24 +1,19 @@ import unittest -from skluc.main.data.mldatasets.Cifar100FineDataset import Cifar100FineDataset from skluc.main.data.mldatasets.Cifar10Dataset import Cifar10Dataset -from skluc.main.data.mldatasets.SVHNDataset import SVHNDataset + from skluc.main.data.transformation.VGG19Transformer import VGG19Transformer + from skluc.main.utils import logger class TestVGG19Transformer(unittest.TestCase): def setUp(self): self.lst_name_cut_layers = [ - "block5_conv4", - "block4_conv4", - "block5_pool", "block3_pool" ] self.dict_datasets = { "cifar10": Cifar10Dataset, - "cifar100": Cifar100FineDataset, - "svhn": SVHNDataset } def test_transform(self): @@ -45,6 +40,7 @@ class TestVGG19Transformer(unittest.TestCase): for name_cut_layer in self.lst_name_cut_layers: logger.info("Testing cut layer {}".format(name_cut_layer)) trans = VGG19Transformer(data_name=data_name, cut_layer_name=name_cut_layer) + logger.debug(trans.keras_model) del trans