Interface TrainingConfig

Model training configuration.

interface TrainingConfig {
    loss: string | string[] | {
        [key: string]: string;
    };
    loss_weights?: number[] | {
        [key: string]: number;
    };
    metrics?: string[] | {
        [key: string]: string;
    };
    optimizer_config: {};
    sample_weight_mode?: string;
    weighted_metrics?: string[];
}

Properties

loss loss_weights? metrics? optimizer_config sample_weight_mode? weighted_metrics?

Properties

loss

loss: string | string[] | {
    [key: string]: string;
}

Loss function(s) for the model's output(s).

Optionalloss_weights

loss_weights?: number[] | {
    [key: string]: number;
}

Optionalmetrics

metrics?: string[] | {
    [key: string]: string;
}

Metric function(s) for the model's output(s).

optimizer_config

optimizer_config: {}

Optimizer used for the model training.

Optionalsample_weight_mode

sample_weight_mode?: string

Optionalweighted_metrics

weighted_metrics?: string[]

Settings

Member Visibility

On This Page

Properties