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

相关推荐
loulanyue_3 分钟前
10000P国产算力破壁上线,算电协同与超智融合打破算力孤岛:高文院士图灵奖得主纵论世界算力网新纪元——读2026世界算力网大会实录
人工智能·大模型·架构设计·高性能计算·算力网络
Thneonl3 分钟前
0.18 也被 DENY:治理必须是台确定性机器
人工智能·架构
老猫聊AI5 分钟前
Qwen-image-2.1 本地部署保姆级教程 | 首页
人工智能·llm
FridaySunny8 分钟前
Agent 工作交接:继续、压缩还是新开?120 次运行的会话策略实验
人工智能
狂潮lister10 分钟前
狂潮 · 元理论v1.0-原创架构设计-AI 网关 | 第9篇:KC-AI‑Gateway(一套可审计的大模型网关完整架构实践)|【完整版】
人工智能·后端
林澈在路上15 分钟前
AI做歌工具怎么拆人声伴奏?分轨导出指南
人工智能·ai音乐·melo音乐
方方洛16 分钟前
ray教程-14-RayTune与超参搜索
人工智能·分布式·机器学习
王吉伟16 分钟前
Anthropic 官方指南:多智能体不等于更强大
人工智能·ai·语言模型·自动化
荆棘鸟智能19 分钟前
扬尘在线监测数据接入与异常清洗:从断点补传到质量审计
人工智能·算法·系统架构
Java内核笔记19 分钟前
同一份知识库,4 种切分 × 3 种召回,纯关键词比纯向量高 13 个点
java·人工智能·spring boot