# - 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 - #