Small changes:

Add test for block5_pool cut layer parameter in test transform
Set slice_size as parameter with default value =100 in utils

skluc/test/test_transformation.py

@@ -9,7 +9,8 @@ class TestVGG19Transformer(unittest.TestCase):
     def setUp(self):
         self.lst_name_cut_layers = [
             "block5_conv4",
-            "block4_conv4"
+            "block4_conv4",
+            "block5_pool"
         ]
         self.lst_offspring_classes = [
             VGG19Cifar10Transformer,

skluc/utils.py

@@ -214,7 +214,7 @@ def replace_nan(tensor):
     return np.where(np.isnan(tensor), np.zeros_like(tensor), tensor)
 
 
-def compute_euristic_sigma_chi2(dataset_full):
+def compute_euristic_sigma_chi2(dataset_full, slice_size=100):
     """
     Given a dataset, return the gamma that should be used (euristically) when using a rbf kernel on this dataset.
 
@@ -224,7 +224,6 @@ def compute_euristic_sigma_chi2(dataset_full):
     :return:
     """
     results = []
-    slice_size = 100
     if slice_size > dataset_full.shape[0]:
         slice_size = dataset_full.shape[0]
     for i in range(dataset_full.shape[0] // slice_size):