模型训练 API

`compile` 方法

Model.compile(
    optimizer="rmsprop",
    loss=None,
    loss_weights=None,
    metrics=None,
    weighted_metrics=None,
    run_eagerly=False,
    steps_per_execution=1,
    jit_compile="auto",
    auto_scale_loss=True,
)

为训练配置模型。

示例

model.compile(
    optimizer=keras.optimizers.Adam(learning_rate=1e-3),
    loss=keras.losses.BinaryCrossentropy(),
    metrics=[
        keras.metrics.BinaryAccuracy(),
        keras.metrics.FalseNegatives(),
    ],
)

参数

optimizer: 字符串（优化器名称）或优化器实例。请参阅 keras.optimizers。
loss: 损失函数。可以是字符串（损失函数名称），或 keras.losses.Loss 实例。请参阅 keras.losses。损失函数是任何具有签名 loss = fn(y_true, y_pred) 的可调用对象，其中 y_true 是真实值，y_pred 是模型的预测值。y_true 应具有形状 (batch_size, d0, .. dN)（稀疏损失函数（如稀疏类别交叉熵）除外，后者期望形状为 (batch_size, d0, .. dN-1) 的整数数组）。y_pred 应具有形状 (batch_size, d0, .. dN)。损失函数应返回一个浮点张量。
loss_weights: 可选的列表或字典，用于指定标量系数（Python 浮点数），以对不同模型输出的损失贡献进行加权。模型将最小化的损失值将是所有单个损失的加权和，由 loss_weights 系数加权。如果是列表，则应与模型的输出具有 1:1 的映射关系。如果是字典，则应将输出名称（字符串）映射到标量系数。
metrics: 在模型训练和测试期间要评估的指标列表。其中每一个都可以是字符串（内置函数的名称）、函数或 keras.metrics.Metric 实例。请参阅 keras.metrics。通常，您将使用 metrics=['accuracy']。函数是任何具有签名 result = fn(y_true, _pred) 的可调用对象。要为多输出模型的不同输出指定不同的指标，您也可以传递字典，例如 metrics={'a':'accuracy', 'b':['accuracy', 'mse']}。您还可以传递一个列表，为每个输出指定一个指标或指标列表，例如 metrics=[['accuracy'], ['accuracy', 'mse']] 或 metrics=['accuracy', ['accuracy', 'mse']]。当您传递字符串 'accuracy' 或 'acc' 时，我们会根据目标和模型输出的形状将其转换为 keras.metrics.BinaryAccuracy、keras.metrics.CategoricalAccuracy、keras.metrics.SparseCategoricalAccuracy 之一。对于字符串 "crossentropy" 和 "ce" 也进行了类似的转换。此处传递的指标在没有样本权重的情况下进行评估；如果您希望应用样本权重，则可以通过 weighted_metrics 参数指定指标。
weighted_metrics: 要在训练和测试期间评估并由 sample_weight 或 class_weight 加权的指标列表。
run_eagerly: 布尔值。如果为 True，则此模型的前向传递将永远不会被编译。建议在训练时将其保留为 False（以获得最佳性能），并在调试时将其设置为 True。
steps_per_execution: 整数。每次编译函数调用期间要运行的批次数。在单个编译函数调用中运行多个批次可以大大提高 TPU 或 Python 开销较大且模型较小的模型的性能。每次执行最多运行一个完整 epoch。如果传递的数字大于 epoch 的大小，则执行将被截断为 epoch 的大小。请注意，如果 steps_per_execution 设置为 N，则 Callback.on_batch_begin 和 Callback.on_batch_end 方法将仅每 N 个批次调用一次（即在每次编译函数执行之前/之后）。PyTorch 后端不支持。
jit_compile: 布尔值或 "auto"。编译模型时是否使用 XLA 编译。对于 jax 和 tensorflow 后端，如果模型支持，jit_compile="auto" 将启用 XLA 编译，否则禁用。对于 torch 后端，"auto" 将默认为 eager 执行，jit_compile=True 将使用 torch.compile 和 "inductor" 后端运行。
auto_scale_loss: 布尔值。如果为 True 且模型 dtype 策略为 "mixed_float16"，则传递的优化器将自动包装在 LossScaleOptimizer 中，这将动态缩放损失以防止下溢。

[来源]

`fit` 方法

Model.fit(
    x=None,
    y=None,
    batch_size=None,
    epochs=1,
    verbose="auto",
    callbacks=None,
    validation_split=0.0,
    validation_data=None,
    shuffle=True,
    class_weight=None,
    sample_weight=None,
    initial_epoch=0,
    steps_per_epoch=None,
    validation_steps=None,
    validation_batch_size=None,
    validation_freq=1,
)

训练模型固定数量的 epoch（数据集迭代）。

参数

x: 输入数据。它可以是
- NumPy 数组（或类数组），或数组列表（如果模型有多个输入）。
- 后端原生张量，或张量列表（如果模型有多个输入）。
- 字典，将输入名称映射到相应的数组/张量，如果模型有命名输入。
- keras.utils.PyDataset，返回 (inputs, targets) 或 (inputs, targets, sample_weights)。
- tf.data.Dataset，产生 (inputs, targets) 或 (inputs, targets, sample_weights)。
- torch.utils.data.DataLoader，产生 (inputs, targets) 或 (inputs, targets, sample_weights)。
- Python 生成器函数，产生 (inputs, targets) 或 (inputs, targets, sample_weights)。
y: 目标数据。与输入数据 x 类似，它可以是 NumPy 数组或后端原生张量。如果 x 是 keras.utils.PyDataset、tf.data.Dataset、torch.utils.data.DataLoader 或 Python 生成器函数，则不应指定 y，因为目标将从 x 获取。
batch_size: 整数或 None。每次梯度更新的样本数。如果未指定，batch_size 将默认为 32。如果输入数据 x 是 keras.utils.PyDataset、tf.data.Dataset、torch.utils.data.DataLoader 或 Python 生成器函数，则不要指定 batch_size，因为它们会生成批次。
epochs: 整数。训练模型的 epoch 数。一个 epoch 是对提供的整个 x 和 y 数据的一次迭代（除非 steps_per_epoch 标志设置为 None 以外的值）。请注意，结合 initial_epoch，epochs 应理解为“最终 epoch”。模型不是训练给定的 epochs 迭代次数，而只是训练到达到索引为 epochs 的 epoch。
verbose: "auto"、0、1 或 2。详细模式。0 = 静默，1 = 进度条，2 = 每 epoch 一行。“auto” 在大多数情况下变为 1。请注意，进度条在记录到文件时不是特别有用，因此在非交互式运行（例如，在生产环境中）时，建议使用 verbose=2。默认为 "auto"。
callbacks: keras.callbacks.Callback 实例列表。训练期间要应用的回调列表。请参阅 keras.callbacks。请注意，keras.callbacks.ProgbarLogger 和 keras.callbacks.History 回调会自动创建，无需传递给 model.fit()。keras.callbacks.ProgbarLogger 是否创建取决于 model.fit() 中的 verbose 参数。
validation_split: 介于 0 和 1 之间的浮点数。用作验证数据的训练数据比例。模型将留出此比例的训练数据，不会对其进行训练，并且将在每个 epoch 结束时评估此数据上的损失和任何模型指标。验证数据是从提供的 x 和 y 数据中的最后样本中选择的，在打乱之前。仅当 x 和 y 由 NumPy 数组或张量组成时，才支持此参数。如果同时提供了 validation_data 和 validation_split，则 validation_data 将覆盖 validation_split。
validation_data: 在每个 epoch 结束时评估损失和任何模型指标的数据。模型不会在此数据上进行训练。因此，请注意，使用 validation_split 或 validation_data 提供的数据的验证损失不受噪声和 dropout 等正则化层的影响。validation_data 将覆盖 validation_split。它可以是
- NumPy 数组或张量的元组 (x_val, y_val)。
- NumPy 数组的元组 (x_val, y_val, val_sample_weights)。
- keras.utils.PyDataset、tf.data.Dataset、torch.utils.data.DataLoader，产生 (inputs, targets) 或 Python 生成器函数，产生 (x_val, y_val) 或 (inputs, targets, sample_weights)。
shuffle: 布尔值，是否在每个 epoch 之前打乱训练数据。当 x 是 keras.utils.PyDataset、tf.data.Dataset、torch.utils.data.DataLoader 或 Python 生成器函数时，将忽略此参数。
class_weight: 可选字典，将类索引（整数）映射到权重（浮点数）值，用于加权损失函数（仅在训练期间）。这对于告诉模型“更多关注”来自代表性不足的类别的样本可能很有用。当指定 class_weight 并且目标的秩为 2 或更大时，y 必须是 one-hot 编码的，或者必须为稀疏类标签包含显式的最终维度 1。
sample_weight: 可选的 NumPy 数组或训练样本权重的张量，用于加权损失函数（仅在训练期间）。您可以传递一个扁平（1D）NumPy 数组或张量，其长度与输入样本的长度相同（权重和样本之间为 1:1 映射），或者在时间数据的情况下，您可以传递一个形状为 (samples, sequence_length) 的 2D NumPy 数组或张量，以将不同的权重应用于每个样本的每个时间步。当 x 是 keras.utils.PyDataset、tf.data.Dataset、torch.utils.data.DataLoader 或 Python 生成器函数时，不支持此参数。相反，请将 sample_weights 作为 x 的第三个元素提供。请注意，样本权重不适用于通过 compile() 中的 metrics 参数指定的指标。要将样本权重应用于指标，您可以通过 compile() 中的 weighted_metrics 指定它们。
initial_epoch: 整数。开始训练的 epoch（对于恢复之前的训练运行很有用）。
steps_per_epoch: 整数或 None。声明一个 epoch 完成并开始下一个 epoch 之前的总步数（样本批次）。当使用输入张量或 NumPy 数组进行训练时，默认值 None 表示使用的值是数据集中的样本数除以批次大小，如果无法确定，则为 1。如果 x 是 keras.utils.PyDataset、tf.data.Dataset、torch.utils.data.DataLoader 或 Python 生成器函数，则 epoch 将运行直到输入数据集耗尽。当传递无限重复的数据集时，您必须指定 steps_per_epoch 参数，否则训练将无限期地运行。
validation_steps: 整数或 None。仅当提供了 validation_data 时才相关。在每个 epoch 结束时执行验证时，停止之前要提取的总步数（样本批次）。如果 validation_steps 为 None，则验证将运行直到 validation_data 数据集耗尽。在无限重复数据集的情况下，它将无限期地运行。如果指定了 validation_steps 并且仅使用了数据集的一部分，则评估将在每个 epoch 开始时从数据集的开头开始。这确保了每次都使用相同的验证样本。
validation_batch_size: 整数或 None。每个验证批次的样本数。如果未指定，将默认为 batch_size。如果您的数据是 keras.utils.PyDataset、tf.data.Dataset、torch.utils.data.DataLoader 或 Python 生成器函数，则不要指定 validation_batch_size，因为它们会生成批次。
validation_freq: 仅当提供验证数据时才相关。指定在执行新的验证运行之前要运行多少个训练 epoch，例如 validation_freq=2 每 2 个 epoch 运行一次验证。

类迭代器输入的解包行为：常见的模式是将类似迭代器的对象（例如 tf.data.Dataset 或 keras.utils.PyDataset）传递给 fit()，实际上不仅会产生特征 (x)，还会可选地产生目标 (y) 和样本权重 (sample_weight)。Keras 要求此类迭代器的输出是明确的。迭代器应返回长度为 1、2 或 3 的元组，其中可选的第二个和第三个元素将分别用于 y 和 sample_weight。提供的任何其他类型都将包装在长度为 1 的元组中，有效地将所有内容都视为 x。当产生字典时，它们仍应遵循顶层元组结构，例如 ({"x0": x0, "x1": x1}, y)。Keras 不会尝试从单个字典的键中分离特征、目标和权重。一个值得注意的不受支持的数据类型是 namedtuple。原因是它的行为既像有序数据类型（元组），又像映射数据类型（字典）。因此，给定形式为 namedtuple("example_tuple", ["y", "x"]) 的 namedtuple，在解释值时是否反转元素的顺序是不明确的。更糟糕的是形式为 namedtuple("other_tuple", ["x", "y", "z"]) 的元组，其中不清楚元组是打算解包到 x、y 和 sample_weight 中，还是作为单个元素传递给 x。

History 对象。其 History.history 属性是连续 epoch 的训练损失值和指标值，以及验证损失值和验证指标值（如果适用）的记录。

[来源]

`evaluate` 方法

Model.evaluate(
    x=None,
    y=None,
    batch_size=None,
    verbose="auto",
    sample_weight=None,
    steps=None,
    callbacks=None,
    return_dict=False,
    **kwargs
)

返回模型在测试模式下的损失值和指标值。

计算以批次完成（请参阅 batch_size 参数）。

参数

x: 输入数据。它可以是
- NumPy 数组（或类数组），或数组列表（如果模型有多个输入）。
- 后端原生张量，或张量列表（如果模型有多个输入）。
- 字典，将输入名称映射到相应的数组/张量，如果模型有命名输入。
- keras.utils.PyDataset，返回 (inputs, targets) 或 (inputs, targets, sample_weights)。
- tf.data.Dataset，产生 (inputs, targets) 或 (inputs, targets, sample_weights)。
- torch.utils.data.DataLoader，产生 (inputs, targets) 或 (inputs, targets, sample_weights)。
- Python 生成器函数，产生 (inputs, targets) 或 (inputs, targets, sample_weights)。
y: 目标数据。与输入数据 x 类似，它可以是 NumPy 数组或后端原生张量。如果 x 是 keras.utils.PyDataset、tf.data.Dataset、torch.utils.data.DataLoader 或 Python 生成器函数，则不应指定 y，因为目标将从 x 获取。
batch_size: 整数或 None。每次计算批次的样本数。如果未指定，batch_size 将默认为 32。如果输入数据 x 是 keras.utils.PyDataset、tf.data.Dataset、torch.utils.data.DataLoader 或 Python 生成器函数，则不要指定 batch_size，因为它们会生成批次。
verbose: "auto"、0、1 或 2。详细模式。0 = 静默，1 = 进度条，2 = 单行。"auto" 在大多数情况下变为 1。请注意，进度条在记录到文件时不是特别有用，因此在非交互式运行（例如，在生产环境中）时，建议使用 verbose=2。默认为 "auto"。
sample_weight: 可选的 NumPy 数组或训练样本权重的张量，用于加权损失函数（仅在训练期间）。您可以传递一个扁平（1D）NumPy 数组或张量，其长度与输入样本的长度相同（权重和样本之间为 1:1 映射），或者在时间数据的情况下，您可以传递一个形状为 (samples, sequence_length) 的 2D NumPy 数组或张量，以将不同的权重应用于每个样本的每个时间步。当 x 是 keras.utils.PyDataset、tf.data.Dataset、torch.utils.data.DataLoader 或 Python 生成器函数时，不支持此参数。相反，请将 sample_weights 作为 x 的第三个元素提供。请注意，样本权重不适用于通过 compile() 中的 metrics 参数指定的指标。要将样本权重应用于指标，您可以通过 compile() 中的 weighted_metrics 指定它们。
steps: 整数或 None。声明评估轮次完成之前要提取的总步数（样本批次）。如果 steps 为 None，则它将运行直到 x 耗尽。在无限重复数据集的情况下，它将无限期地运行。
callbacks: keras.callbacks.Callback 实例列表。评估期间要应用的回调列表。
return_dict: 如果为 True，则损失和指标结果将作为字典返回，其中每个键都是指标的名称。如果为 False，则它们将作为列表返回。

标量测试损失（如果模型具有单个输出且没有指标）或标量列表（如果模型具有多个输出和/或指标）。属性 model.metrics_names 将为您提供标量输出的显示标签。

[来源]

`predict` 方法

Model.predict(x, batch_size=None, verbose="auto", steps=None, callbacks=None)

为输入样本生成输出预测。

计算以批次完成。此方法旨在批量处理大量输入。它不适用于在循环内迭代数据并一次处理少量输入。

对于适合一个批次的少量输入，直接使用 __call__() 以获得更快的执行速度，例如 model(x)，或者如果您的层（如 BatchNormalization）在推理期间的行为不同，则使用 model(x, training=False)。

注意：有关 Model 方法 predict() 和 __call__() 之间差异的更多详细信息，请参阅此 FAQ 条目。

参数

x: 输入数据。它可以是
- NumPy 数组（或类数组），或数组列表（如果模型有多个输入）。
- 后端原生张量，或张量列表（如果模型有多个输入）。
- 字典，将输入名称映射到相应的数组/张量，如果模型有命名输入。
- keras.utils.PyDataset。
- tf.data.Dataset。
- torch.utils.data.DataLoader。
- Python 生成器函数。
batch_size: 整数或 None。每次计算批次的样本数。如果未指定，batch_size 将默认为 32。如果输入数据 x 是 keras.utils.PyDataset、tf.data.Dataset、torch.utils.data.DataLoader 或 Python 生成器函数，则不要指定 batch_size，因为它们会生成批次。
verbose: "auto"、0、1 或 2。详细模式。0 = 静默，1 = 进度条，2 = 单行。"auto" 在大多数情况下变为 1。请注意，进度条在记录到文件时不是特别有用，因此在非交互式运行（例如，在生产环境中）时，建议使用 verbose=2。默认为 "auto"。
steps: 声明预测轮次完成之前要提取的总步数（样本批次）。如果 steps 为 None，则它将运行直到 x 耗尽。在无限重复数据集的情况下，它将无限期地运行。
callbacks: keras.callbacks.Callback 实例列表。预测期间要应用的回调列表。

预测的 NumPy 数组。

[来源]

`train_on_batch` 方法

Model.train_on_batch(
    x, y=None, sample_weight=None, class_weight=None, return_dict=False
)

对单个数据批次运行单个梯度更新。

参数

x: 输入数据。必须是类数组。
y: 目标数据。必须是类数组。
sample_weight: 可选的与 x 长度相同的数组，包含要应用于每个样本的模型损失的权重。在时间数据的情况下，您可以传递形状为 (samples, sequence_length) 的 2D 数组，以将不同的权重应用于每个样本的每个时间步。
class_weight: 可选字典，将类索引（整数）映射到权重（浮点数），以应用于训练期间来自此类的样本的模型损失。这对于告诉模型“更多关注”来自代表性不足的类别的样本可能很有用。当指定 class_weight 并且目标的秩为 2 或更大时，y 必须是 one-hot 编码的，或者必须为稀疏类标签包含显式的最终维度 1。
return_dict: 如果为 True，则损失和指标结果将作为字典返回，其中每个键都是指标的名称。如果为 False，则它们将作为列表返回。

标量损失值（当没有指标且 return_dict=False 时）、损失和指标值列表（如果有指标且 return_dict=False），或指标和损失值字典（如果 return_dict=True）。

[来源]

`test_on_batch` 方法

Model.test_on_batch(x, y=None, sample_weight=None, return_dict=False)

在单个样本批次上测试模型。

参数

x: 输入数据。必须是类数组。
y: 目标数据。必须是类数组。
sample_weight: 可选的与 x 长度相同的数组，包含要应用于每个样本的模型损失的权重。在时间数据的情况下，您可以传递形状为 (samples, sequence_length) 的 2D 数组，以将不同的权重应用于每个样本的每个时间步。
return_dict: 如果为 True，则损失和指标结果将作为字典返回，其中每个键都是指标的名称。如果为 False，则它们将作为列表返回。

标量损失值（当没有指标且 return_dict=False 时）、损失和指标值列表（如果有指标且 return_dict=False），或指标和损失值字典（如果 return_dict=True）。

[来源]

`predict_on_batch` 方法

Model.predict_on_batch(x)

返回单个样本批次的预测。

参数

x: 输入数据。它必须是类数组。

预测的 NumPy 数组。

模型训练 API

compile 方法

fit 方法

evaluate 方法

predict 方法

train_on_batch 方法

test_on_batch 方法

predict_on_batch 方法

模型训练 API

compile 方法

fit 方法

evaluate 方法

predict 方法

train_on_batch 方法

test_on_batch 方法

predict_on_batch 方法

`compile` 方法

`fit` 方法

`evaluate` 方法

`predict` 方法

`train_on_batch` 方法

`test_on_batch` 方法

`predict_on_batch` 方法