def train()

in src/deep_demand_forecast/train.py [0:0]

• 15 lines of code
• 1 McCabe index (conditional complexity)

def train(
    dataset: TrainDatasets,
    output_dir: str,
    model_dir: str,
    context_length: int,
    prediction_length: int,
    skip_size: int,
    ar_window: int,
    channels: int,
    scaling: bool,
    output_activation: str,
    epochs: int,
    batch_size: int,
    learning_rate: float,
    seed: int,