美文网首页
训练并评估用函数式API创建的模型

训练并评估用函数式API创建的模型

作者: LabVIEW_Python | 来源:发表于2021-09-22 09:26 被阅读0次

训练并评估用函数式API创建的模型,跟顺序模型的训练和评估基本一致:

  • 载入并预处理数据
  • 用compile方法配置模型:loss/optimizer/metrics
  • 用fit方法训练模型,实现模型"适配"数据,返回训练损失值和指标值的记录
  • 用evaluate方法在测试数据集上测试模型,返回 list:[loss, metrics]
    完整范例代码如下所示:
import tensorflow as tf 
from tensorflow.keras import layers
from tensorflow import keras 

# 创建层,将上一层传递进当前层
input = keras.Input(shape=(784,))
dense = layers.Dense(64,activation='relu')
x1 = dense(input)
x2 = layers.Dense(32, activation='relu')(x1)
output = layers.Dense(10)(x2)

# 通过输入\输出创建模型
model = keras.Model(inputs=input, outputs=output, name="Alex_Model")

# 查看模型摘要
model.summary()

# 载入并预处理数据
(x_train, y_train),(x_test, y_test) = keras.datasets.mnist.load_data()
x_train = x_train.reshape(60000,-1).astype("float32")/255.0
x_test  = x_test.reshape(10000,-1).astype("float32")/255.0

# 用compile方法配置模型:loss/optimizer/metrics
model.compile(optimizer="rmsprop", loss=keras.losses.SparseCategoricalCrossentropy(from_logits=True),metrics=["accuracy"])

# 用fit方法训练模型,实现模型"适配"数据,返回训练损失值和指标值的记录
history = model.fit(x_train,y_train, batch_size=64, epochs=3, validation_split=0.2)
print(history.history.keys()) # dict_keys(['loss', 'accuracy', 'val_loss', 'val_accuracy'])

# 用evaluate方法在测试数据集上测试模型,返回 list:[loss, metrics]
test_scores = model.evaluate(x_test, y_test, verbose=2)
print("Test loss:", test_scores[0])
print("Test accuracy:", test_scores[1])

相关文章

  • 训练并评估用函数式API创建的模型

    训练并评估用函数式API创建的模型,跟顺序模型的训练和评估基本一致: 载入并预处理数据 用compile方法配置模...

  • 用Keras的函数式API创建模型

    为什么要用Keras的函数式API创建模型? 支持创建多输入多输出模型 支持创建非线性模型,eg: 残差模块 支持...

  • TensorFlow 编程的基本流程

    1. 数据集 2. 创建模型 3. 训练 4. 评估模型 5. 预测 附:TensorFlow API 层的编程堆栈

  • keras

    Keras设计了俩种构建模型的方式函数式模型API和顺序式模型API 顺序式模型API构建模型示例: from k...

  • Keras函数式 API

    Keras 函数式 API 是定义复杂模型(如多输出模型、有向无环图,或具有共享层的模型)的方法。 非函数式api...

  • Keras多输出模型构建

    1、多输出模型 使用keras函数式API构建网络: 2、自定义loss函数 3、批量训练 4、调试 在自定义的l...

  • tf教程4: 因果卷积+LSTM预测时序数据

    导入所需的包 构造人工时序数据 辅助函数 创建模型并训练 选择合适的学习率 使用优化后的学习率重新训练 使用模型预...

  • 81-mlr3初体验

    1、创建任务 2、选择学习器 3、拆分训练集和测试集 4、训练模型 5、预测 6、模型评估 7、交叉验证

  • GATK的VQSR介绍

    高斯混合模型 使用高斯混合模型创建训练集,根据该训练集评估每个变异位点的可信度。每次运行VariantRecali...

  • 构建高级模型(05)

    函数式 API tf.keras.Sequential 模型是层的简单堆叠,无法表示任意模型。使用 Keras 函...

网友评论

      本文标题:训练并评估用函数式API创建的模型

      本文链接:https://www.haomeiwen.com/subject/huuwgltx.html