tensorflow2中fit函数

在 TensorFlow 2 中，`fit()` 函数是训练模型的核心函数之一，它通常与 `compile()` 函数和 `predict()` 函数一起使用。`fit()` 函数可以用于训练多种不同类型的模型，如深度神经网络、卷积神经网络和循环神经网络等。

下面是一个简单的使用 `fit()` 函数训练模型的示例：

import tensorflow as tf

# 准备数据
(x_train, y_train), (x_test, y_test) = tf.keras.datasets.mnist.load_data()
x_train = x_train.reshape(x_train.shape[0], 784)
x_test = x_test.reshape(x_test.shape[0], 784)
x_train = x_train.astype('float32')
x_test = x_test.astype('float32')
x_train /= 255
x_test /= 255

# 定义模型
model = tf.keras.models.Sequential([
    tf.keras.layers.Dense(128, activation='relu'),
    tf.keras.layers.Dense(10, activation='softmax')
])

# 编译模型
model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

# 训练模型
model.fit(x_train, y_train, epochs=5, validation_data=(x_test, y_test))

在这个例子中，我们首先加载了 MNIST 数据集，并对数据进行了预处理。然后，我们定义了一个简单的全连接神经网络模型，并使用 `compile()` 函数指定了优化器、损失函数和评价指标。最后，我们使用 `fit()` 函数对模型进行了训练，并指定了训练的轮数和验证集。

`fit()` 函数还支持许多其他参数，例如 batch_size、callbacks 和 class_weight 等，这些参数可以根据具体的需求进行调整。