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

相关推荐
EllinY4 分钟前
CF2217E Definitely Larger 题解
c++·笔记·算法·构造
Raink老师7 分钟前
【AI面试临阵磨枪-72】电商全场景 AI Agent 设计(商品咨询 / 订单 / 物流 / 售后 / 退款)
人工智能·面试·职场和发展
仙女修炼史14 分钟前
CNN更看重Texture还是shape:imagenet-trained cnns are biased
论文阅读·人工智能·cnn
nnsix16 分钟前
JIT(即时编译)、 AOT(预先编译)、虚拟机CLR 笔记
笔记
视***间30 分钟前
视程空间 AIR SC6N0-C-MB NX 16GB 规格详解与机器人/机器狗适配说明
人工智能·机器人·边缘计算·机器狗·ai算力·具身机器人·视程空间
ZC跨境爬虫30 分钟前
跟着 MDN 学CSS day_16:(深入掌握背景与边框的艺术)
前端·css·ui·html·tensorflow
视***间31 分钟前
小身板・强算力・全适配 —— 视程空间 AI 算力开发板如何完美适配机器人 / 机器狗
人工智能·机器人·边缘计算·ai算力·视程空间·算力开发板
网宿安全演武实验室34 分钟前
当AI跑进容器:全链路容器安全检测与智能运营实
人工智能·安全·容器·k8s
Cosolar36 分钟前
2026年AI Agent技术生态开源项目合集
人工智能·开源·agent·智能体
白日做梦Q40 分钟前
Label Studio 安装与使用完整文档(可直接复制部署)
深度学习·yolo·计算机视觉