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