Fix bug in dataset_postprocessor in which the test dataset was generated from the train dataset

ruancomelli · ruancomelli · commit bc2776bd02fc · 2021-06-17T15:03:07.000-03:00
diff --git a/boiling_learning/datasets/creators.py b/boiling_learning/datasets/creators.py
@@ -240,7 +240,7 @@ def dataset_post_processor(
                     data_augmentor.as_tf_py_function(pack_tuple=True),
                     num_parallel_calls=AUTOTUNE,
                 )
-            ds_test = ds_train.map(
+            ds_test = ds_test.map(
                 data_augmentor.as_tf_py_function(pack_tuple=True),
                 num_parallel_calls=AUTOTUNE,
             )
@@ -263,4 +263,4 @@ def dataset_post_processor(
             ds_val = ds_val.prefetch(AUTOTUNE)
         ds_test = ds_test.prefetch(AUTOTUNE)
 
-    return (ds_train, ds_val, ds_test)
+    return ds_train, ds_val, ds_test

Original file line number	Diff line number	Diff line change
`@@ -240,7 +240,7 @@ def dataset_post_processor(`
`240`	`240`	`data_augmentor.as_tf_py_function(pack_tuple=True),`
`241`	`241`	`num_parallel_calls=AUTOTUNE,`
`242`	`242`	`)`
`243`		`- ds_test = ds_train.map(`
	`243`	`+ ds_test = ds_test.map(`
`244`	`244`	`data_augmentor.as_tf_py_function(pack_tuple=True),`
`245`	`245`	`num_parallel_calls=AUTOTUNE,`
`246`	`246`	`)`
`@@ -263,4 +263,4 @@ def dataset_post_processor(`
`263`	`263`	`ds_val = ds_val.prefetch(AUTOTUNE)`
`264`	`264`	`ds_test = ds_test.prefetch(AUTOTUNE)`
`265`	`265`
`266`		`- return (ds_train, ds_val, ds_test)`
	`266`	`+ return ds_train, ds_val, ds_test`