- 博客(2)
- 收藏
- 关注
RSS订阅原创 一个简单的神经网络项目（新冠感染预测）完整流程
import csv为什么使用torch？torch支持张量在GPU运行，速度快，效率高。一般的npm数组只能在CPU上计算。所以想要使用torch，需要支持CUDA的NVIDIA显卡。总结就是数据准备和加载采用npm数组（在CPU），数据训练时采用张量（在GPU），最后可视化和保存的时候又要从GPU转移到CPUimport csv是用来处理 CSV（Comma-Separated Values）文件的Python标准库模块，可以从Excel导出的CSV文件中读取数据。
2026-01-21 14:19:26
879
原创 训练获得线性拟合图
w_0 = torch.normal(0, 0.01, true_w.shape, requires_grad=True) #这个w需要计算梯度。torch.normal参数中，第一个为均值，第二个为标准差，第三个为张量的形状，如（500，4）表示有500个例子且每个例子有4个特征。这里可以根据画图包观察一下得到的数据对应的散点图，由于是4个x，对于x-y坐标系，每次只能提供一个x与y的对应关系。plt.scatter(X[:, 3], Y, 1)#取特征值矩阵第4列与Y的对应关系，像素大小为1。
2026-01-19 15:44:30
366
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人