模型编译

简介: 【8月更文挑战第20天】模型编译。

模型编译,确定训练流程
构建好模型后,通过调用compile配置该模型的学习流程:
compile( optimizer='rmsprop', loss=None, metrics=None, loss_weights=None):
optimizer:优化器;
loss:损失函数,对于二分类任务就是交叉熵,回归任务就是 mse 之类的;
metrics:在训练和测试期间的模型评估标准。比如 metrics = ['accuracy']。 指定不同的评估标准,需要传递一个字典,如 metrics = {'output_a':'accuracy'}。
loss_weights: 如果的模型有多个任务输出,在优化全局loss的时候,需要给每个输出指定相应的权重。

代码:
model = tf.keras.Sequential()
model.add(layers.Dense(10, activation='softmax'))

确定优化器(optimizer)、损失函数(loss)、模型评估方法(metrics)

model.compile(optimizer=tf.keras.optimizers.Adam(0.001),
loss=tf.keras.losses.categorical_crossentropy,
metrics=[tf.keras.metrics.categorical_accuracy])

相关文章
C++ 一种在编译阶段就能解决代码的技术
C++ 一种在编译阶段就能解决代码的技术
C编译环境和预处理(非常详细,建议收藏)
C编译环境和预处理(非常详细,建议收藏)
109 0
不惧编译error,编译Tensorflow源码(二)
编译tensorflow遇到的bug本来就多,在Windows平台上bugs更是加大力度。明明官方教程中在配置完环境后只需执行两行bazel命令,第一行命令却产生不少error。笔者踩了不少坑后,总结出了一些解决方法形成此教程。
不惧编译error,编译Tensorflow源码(二)
不惧编译error,编译Tensorflow源码(一)
编译tensorflow遇到的bug本来就多,在Windows平台上bugs更是加大力度。明明官方教程中在配置完环境后只需执行两行bazel命令,第一行命令却产生不少error。笔者踩了不少坑后,总结出了一些解决方法形成此教程。
不惧编译error,编译Tensorflow源码(一)
AI助理

你好,我是AI助理

可以解答问题、推荐解决方案等