微信公众号:文艺数学君
前言
目前网络上关于深度学习框架的教程有很多,我自己也是看了一些,但是比较零散,于是想在这里把关于深度学习框架的教程整理一下。
目前这篇文章中包括的框架有TensorFlow,PyTorch和Keras。这一篇文章会使用这三种框架来实现以下线性回归。做线性回归,既可以顺一遍整个框架的流程,也不会太麻烦。
视频
这个视频和主题没什么关系吧,拍的校园里的一角,微信压缩比较厉害,视频好像比较模糊了。
深度学习框架介绍
实验数据生成
生成的图片如下所示:
接下来就是使用三种框架进行实现回归了。
TensorFlow
我们可以得到下面的输出结果,上面代码注释很明白了,就不多做解释了。
Keras
同样,使用这个框架可以得到如下的输出结果,看你和的曲线效果还是很不错的:
PyTorch
我们可以看一下PyTorch框架画出的图像:
结语
使用深度学习的框架来实现线性回归有点是大炮打蚊子,但是用这个例子来熟悉这三个框架还是很不错的,代码段不是很长,数据好少,训练时间不长,又能有一遍完整的流程,还是很不错的。
关于更多的内容可以通过文艺数学君的网站进行获取。
领取专属 10元无门槛券
私享最新 技术干货