Mercurial > repos > siwaa > wine_quality_train_eval

comparison toto_wine_quality_train_eval.py @ 11:b5f69f836e03 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://forgemia.inra.fr/nathalie.rousse/use/-/tree/dnn/DNN/galaxy-tools/wine_quality_train_eval commit e7fd13c34ec074a7ebc246301b5a80069dcbcc3a-dirty
author siwaa
date Thu, 05 Dec 2024 15:55:00 +0000
parents b432386c0f1c
children 60778af2dd78
comparison
equal deleted inserted replaced
10:b432386c0f1c 11:b5f69f836e03
96 96
97 # Import some packages 97 # Import some packages
98 import os 98 import os
99 import lightning.pytorch as pl 99 import lightning.pytorch as pl
100 import torchvision.transforms as T 100 import torchvision.transforms as T
101 from IPython.display import display, HTML ##toto## HTML 101 ##toto## from IPython.display import display, HTML ##toto## HTML
102 from torch.utils.data import DataLoader, random_split 102 from torch.utils.data import DataLoader, random_split
103 from model_wine_lightning.modules.progressbar import CustomTrainProgressBar 103 from model_wine_lightning.modules.progressbar import CustomTrainProgressBar
104 from model_wine_lightning.modules.data_load import WineQualityDataset 104 from model_wine_lightning.modules.data_load import WineQualityDataset
105 from model_wine_lightning.modules.data_load import Normalize, ToTensor 105 from model_wine_lightning.modules.data_load import Normalize, ToTensor
106 from model_wine_lightning.modules.model import LitRegression 106 from model_wine_lightning.modules.model import LitRegression