--- a/toto_wine_quality_train_eval.py	Thu Dec 05 15:40:01 2024 +0000
+++ b/toto_wine_quality_train_eval.py	Thu Dec 05 15:47:56 2024 +0000
@@ -98,7 +98,7 @@
 import os
 import lightning.pytorch as pl
 import torchvision.transforms as T
-from IPython.display import display, HTML
+from IPython.display import display, HTML ##toto## HTML
 from torch.utils.data import DataLoader, random_split
 from model_wine_lightning.modules.progressbar import CustomTrainProgressBar
 from model_wine_lightning.modules.data_load import WineQualityDataset
@@ -186,7 +186,7 @@
     datasets = WineQualityDataset(dataset_filepath)
     print("datasets:")
     #display(datasets.data.head(5).style.format("{0:.2f}"))
-    display(datasets.data.head(5))
+    ##toto##display(datasets.data.head(5))
     print('Missing Data : ',datasets.data.isna().sum().sum(),
           '  Shape is : ', datasets.data.shape)
 
@@ -209,9 +209,9 @@
     dataset = WineQualityDataset(dataset_filepath, transform=transforms)
 
     print("Before normalization :")
-    display(datasets[:]["features"])
+    ##toto##display(datasets[:]["features"])
     print("After normalization :")
-    display(dataset[:]["features"])
+    ##toto##display(dataset[:]["features"])
 
     # ### 3.2 - Split data
     print("\n"+HEAD,"# ### 3.2 - Split data\n")