@tensorflow/tfjs-layers/dist/keras_format/training_config.d.ts

TensorFlow layers API in JavaScript

/// <amd-module name="@tensorflow/tfjs-layers/dist/keras_format/training_config" />
/**
 * @license
 * Copyright 2018 Google LLC
 *
 * Use of this source code is governed by an MIT-style
 * license that can be found in the LICENSE file or at
 * https://opensource.org/licenses/MIT.
 * =============================================================================
 */
import { SampleWeightMode } from './common';
import { LossIdentifier } from './loss_config';
import { OptimizerSerialization } from './optimizer_config';
import { PyJsonDict } from './types';
export type MetricsIdentifier = string;
/**
 * a type for valid values of the `loss_weights` field.
 */
export type LossWeights = number[] | {
    [key: string]: number;
};
/**
 * Configuration of the Keras trainer. This includes the configuration to the
 * optimizer, the loss, any metrics to be calculated, etc.
 */
export interface TrainingConfig extends PyJsonDict {
    optimizer_config: OptimizerSerialization;
    loss: LossIdentifier | LossIdentifier[] | {
        [key: string]: LossIdentifier;
    };
    metrics?: MetricsIdentifier[] | {
        [key: string]: MetricsIdentifier;
    };
    weighted_metrics?: MetricsIdentifier[];
    sample_weight_mode?: SampleWeightMode;
    loss_weights?: LossWeights;
}