TensorFlow
tf.keras API Reference
tf.keras 是 TensorFlow 的高层 API,提供层、模型、训练与数据管道抽象。
By EZ4Code Team
Keras API
顺序模型、网络层、编译训练与数据集构建的核心 API。
tf.keras.Sequential(layers)创建由若干层线性堆叠的顺序模型。
Returns: Sequential — 顺序模型
tf.keras.layers.Dense(units, activation)创建全连接层。
Returns: Dense — 全连接层
tf.keras.layers.Conv2D(filters, kernel_size)创建二维卷积层,用于图像特征提取。
Returns: Conv2D — 二维卷积层
model.compile(optimizer, loss, metrics)配置模型的优化器、损失函数与评估指标。
Returns: None — 配置模型(原地)
model.fit(x, y, epochs, batch_size)用训练数据训练模型若干轮。
Returns: History — 训练历史记录
model.evaluate(x, y)在测试数据上评估模型损失与指标。
Returns: list — 损失与各指标的值
model.predict(x)对输入数据进行前向推理,返回预测结果。
Returns: ndarray — 预测输出
tf.data.Dataset.from_tensor_slices(data)从内存中的张量创建数据集,支持批处理、打乱与预取。
Returns: Dataset — tf.data 数据集