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

相关推荐
zhippt_10145 分钟前
只剩截图没有源文件?图转PPT把图片还原成可编辑幻灯片
前端·人工智能·powerpoint
wuyoutrans9 分钟前
以色列专利OA答复翻译,重点要避开哪些坑?
笔记
小oo呆10 分钟前
【自然语言处理与大模型】自然语言处理的漫漫长路之深度学习时代
人工智能·深度学习·自然语言处理
宝绿特塑业再生科技14 分钟前
HDPE再生颗粒户外家具表面粉化机理与改性技术全解析
笔记·户外家具·再生塑料·塑木桌椅
JavaPub-rodert17 分钟前
AI Agent 的记忆系统怎么设计?从短期记忆、长期记忆到 RAG,讲透 Memory 工程化
人工智能
智恒百亿18 分钟前
一台 5090 八卡服务器怎么分给一个团队用?多模型多任务的显存分配实战
java·大数据·人工智能
MCT12365421 分钟前
AI VRM 高频设计|功率电感磁材选型与电源损耗、效率关系
人工智能·dc-dc电源·功率电感·一体成型电感
汤姆宝库23 分钟前
安全深拆:越狱、Sleeper与可解释性
人工智能
skywalk816327 分钟前
lightmem记忆系统实战存取.小浣熊一句话:从lightmem记忆中,取出FreeBSD linux仿真环境安装启动workbuddy的信息,整理成文档
人工智能·硬件架构·记忆·lightmem
Solange__27 分钟前
在线3D预览器、Blender和游戏引擎怎么选?按5类任务比较格式、隐私与编辑性
人工智能·游戏·3d·游戏引擎·blender·1024程序员节·关卡设计