模型编译

简介: 【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++ 元对象系统03】深入探索Qt反射:从原理到实践
【C++ 元对象系统03】深入探索Qt反射:从原理到实践
599 5
|
存储 缓存 异构计算
大语言模型量化方法对比:GPTQ、GGUF、AWQ
在过去的一年里,大型语言模型(llm)有了飞速的发展,在本文中,我们将探讨几种(量化)的方式,除此以外,还会介绍分片及不同的保存和压缩策略。
5392 0
|
7月前
|
机器学习/深度学习 TensorFlow 算法框架/工具
YOLOv11改进策略【注意力机制篇】| 引入MobileNetv4中的Mobile MQA,轻量化注意力模块 提高模型效率
YOLOv11改进策略【注意力机制篇】| 引入MobileNetv4中的Mobile MQA,轻量化注意力模块 提高模型效率
397 0
|
10月前
|
消息中间件 存储 安全
|
Linux C语言
Linux: 僵尸进程究竟是什么?有什么危害?
Linux: 僵尸进程究竟是什么?有什么危害?
258 0
|
存储 机器学习/深度学习 测试技术
模型量化技术综述:揭示大型语言模型压缩的前沿技术
在这篇文章中,我将在语言建模的背景下介绍量化,并逐一探讨各个概念,探索各种方法论、用例以及量化背后的原理。
375 0
模型量化技术综述:揭示大型语言模型压缩的前沿技术
|
存储 运维 监控
复星健康集团陈清阳:云边端协同一体化,构建新一代双SaaS星医平台
上海复星健康科技(集团)有限公司技术总监陈清阳在【一云多形态】专场中带来了题为《云边端协同一体化,构建新一代双SaaS星医平台》的主题演讲,围绕复星健康集团在双SaaS服务建设过程中选型思考、集团数据能力建设、混合语音组网解决方案、复星健康的平台化工程实践等相关话题展开分享。
|
算法 安全 API
Zipline 3.0 中文文档(二)(2)
Zipline 3.0 中文文档(二)
203 3
|
监控 网络协议 网络安全
【专栏】RTMP和RTSP是流媒体传输常用的协议:秒懂
【4月更文挑战第28天】RTMP和RTSP是流媒体传输常用的协议。RTMP由Adobe开发,适合低延迟的实时通信,常用于网络直播和在线游戏;而RTSP是IETF定义的协议,侧重于流媒体播放控制,适用于视频监控和VoD服务。RTMP在业界普及度高,RTSP则在专业领域更常见。选择时需考虑延迟、应用场景和安全性等因素。
1722 1
经验大分享:Sched_Boost小结
经验大分享:Sched_Boost小结
198 0