跟无神学AI之Tensorflow笔记搭建网络八股

虽然Pytorch在论文中使用较多,但是像Alphafold在蛋白质结构预测的模型,仍然是用Tensorflow写成,遂近期在学其中的语法。

本系列来自慕课北大软微曹健老师的Tensorflow笔记,摘选其中重要部分。

1.导包

2.定义训练集测试集和数据表示。

3.用Sequential将模型的层整合起来,也就是定义模型全部层。

拉直层Flatten不改变参数,只是特征的形状转化,变成一维。

此时注意,Sequential方法不能搭建有跳连结构的,需要使用类方法,创建模型类,如下:

类方法中的Model指的是继承Tensorflow中的model类。

Dense()为全连接层,即其中的D1块是一个3层的全连接网络,创建类在Tensorflow中的风格一样。

4.在compile中指定优化器,损失函数和评测指标。

可以选用字符串形式或者函数形式。

其中,from_logits如果等于False指的是输出结果经过概率分布的处理。

5.在fit中进行训练过程,告知训练过程:测试集/训练集的输入特征、标签,batch、迭代次数。

6.用summary打印出参数统计和网络结果。

以上六步操作是使用Tensorflow进行网络搭建的提纲

给出的一个鸢尾花分类六部法:后者是创建类的模型,一般后者肯定更常用,复杂的模型不能一定的不可能全是前向神经网络。

觉得有用的话,欢迎给个一键三连哦,关注无神一起学AI!

相关推荐
小程故事多_8010 分钟前
从A2C、TRPO、PPO到GRPO,强化学习策略梯度算法完整演进与大模型落地实战解析
人工智能·算法
风曦Kisaki15 分钟前
Kubernetes(K8s)笔记Day06: 持久化存储(emptyDir,hostPath,NFS),PV 和 PVC,StorageClass存储类
linux·运维·笔记·云原生·容器·kubernetes
冬奇Lab16 分钟前
开源项目第176期:Better Harness — 不审查 diff,审查工作流本身,给 AI 编程 Agent 的五维评估框架
人工智能·开源·agent
冬奇Lab22 分钟前
代码库知识库系列(07):混合检索 BM25 + 向量——Q8 还是失败,而且总分退步了
人工智能
ajassi200037 分钟前
AI语音智能体开发日记(十一)为智能设备“声”临其境——详解音频资源自动化生成流程
人工智能·ai·ai编程
2601_949499941 小时前
400G组网低功耗优选!芯瑞科技400G VR4 QSFP112光模块赋能智算中心高速互联
大数据·人工智能·科技
GoAI1 小时前
# AI Agent 记忆框架横向对比报告总结
人工智能·大模型·llm·多模态
AI人工智能+1 小时前
一种基于深度学习技术的高精度医疗机构执业许可证识别系统,构建了一套基于深度神经网络的端到端智能识别系统,为医疗行业提
深度学习·ocr·医疗机构执业许可证识别
硅谷秋水2 小时前
EgoSteer:一种基于第一人称视角视频、实现可控灵巧操作的全栈系统
深度学习·机器学习·语言模型·机器人·音视频
李昊哲小课2 小时前
fastapi sse websocket 奶茶店实时订单看板
人工智能·python·websocket·网络协议·fastapi·sse