Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

使用残差神经网络实现一个分类问题,家人们

图片数据集不好找,我们直接使用随机生成的一些数来当作数据集和测试集

因为生成的数都是单独的也就是1*1的所以咱们的分类任务足够简单只使用全连接层就够用

残差块是三层全连接

主网络由三个残差块组成

那就实现一个奇数和偶数的分类问题吧

介绍一下文件奥 main是主函数文件,训练模型和测试 NetWork是agen和网络的类 model文件夹存的当然是模型了

因为最开始发现只输入一个数进行训练不行

因为只有一个数的话特征太少了,对于网络来说没啥特征可以找

所以我把每一个数都换成了二进制文件传入智能体进行训练

结果表明,只需要十次训练就可以达到100%,效果很好

因为数据很少所有就没有写GPU训练,直接使用默认的cpu进行训练了

About

No description, website, or topics provided.

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages