File size: 984 Bytes
dbd79bd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# - x - x - x - x - x - x - x - x - x - x - x - x - x - x - #
#                                                           #
#   This file was created by: Alberto Palomo Alonso         #
# Universidad de Alcalá - Escuela Politécnica Superior      #
#                                                           #
# - x - x - x - x - x - x - x - x - x - x - x - x - x - x - #
# Import statements:
from dataclasses import dataclass


@dataclass
class DatasetConfig:
    # Paths:
    train_data_path: str = None
    val_data_path: str = None
    test_data_path: str = None
    # Percentages:
    train_percentage: float = 1.0
    val_percentage: float = 1.0
    test_percentage: float = 1.0
    # Other parameters:
    num_workers: int = 0
    shuffle_train: bool = True
    shuffle_val: bool = True


# - x - x - x - x - x - x - x - x - x - x - x - x - x - x - #
#                        END OF FILE                        #
# - x - x - x - x - x - x - x - x - x - x - x - x - x - x - #