跟无神学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!

相关推荐
算家计算几秒前
编程AI新王Claude Opus 4.5正式发布!编程基准突破80.9%,成本降三分之二
人工智能·ai编程·claude
青瓷程序设计17 分钟前
鱼类识别系统【最新版】Python+TensorFlow+Vue3+Django+人工智能+深度学习+卷积神经网络算法
人工智能·python·深度学习
央链知播20 分钟前
第二届中国数据产业发展大会暨2025元宇宙AI数据要素“金杏奖”颁奖盛典在广州隆重举行
人工智能·业界资讯·数据产业
GEO_NEWS35 分钟前
解析华为Flex:ai的开源棋局
人工智能·华为·开源
扑棱蛾子43 分钟前
手摸手教你两分钟搞定Antigravity
人工智能
WWZZ20251 小时前
快速上手大模型:深度学习13(文本预处理、语言模型、RNN、GRU、LSTM、seq2seq)
人工智能·深度学习·算法·语言模型·自然语言处理·大模型·具身智能
老友@1 小时前
RAG 的诞生:为了让 AI 不再“乱编”
人工智能·搜索引擎·ai·语言模型·自然语言处理·rag
三条猫1 小时前
将3D CAD 模型结构树转换为图结构,用于训练CAD AI的思路
人工智能·3d·ai·cad·模型训练·图结构·结构树
攻城狮7号1 小时前
Meta开源SAM 3D,如何教会 AI “想象”三维世界
人工智能·开源大模型·sam 3d·2d变3d·meta大模型·3d 视觉ai
三七互娱后端团队1 小时前
告别“玄学”调参:DSPy 框架入门,让 AI 自动优化 AI 的提示词
人工智能·后端