# Autogenerated code. DO NOT EDIT.
import py_gen_ml as pgml

from . import advanced_base as base


class LinearBlockPatch(pgml.YamlBaseModel):
    """Linear block configuration"""

    num_units: int | None = None
    """Number of units"""

    activation: str | None = None
    """Activation function"""



class OptimizerPatch(pgml.YamlBaseModel):
    """Optimizer configuration"""

    type: str | None = None
    """Type of optimizer"""

    learning_rate: float | None = None
    """Learning rate"""



class MLPPatch(pgml.YamlBaseModel):
    """Multi-layer perceptron configuration"""

    layers: list[LinearBlockPatch] | None = None
    """List of linear blocks"""



class TrainingPatch(pgml.YamlBaseModel):
    """Training configuration"""

    mlp: MLPPatch | None = None
    """Multi-layer perceptron configuration"""

    optimizer: OptimizerPatch | None = None
    """Optimizer configuration"""


