文件名称:LinearRegression
文件大小:15KB
文件格式:ZIP
更新时间:2024-04-16 08:57:07
Python
PyTorch线性回归PyTorch实现线性回归 包括模型训练,保存,推理使用等 输入输出维度可以都是一维的 torch.nn.Linear(1, 1) 输入可以多维输出可以是一维的 torch.nn.Linear(10, 1) 整个文件夹结构如下 ├── checkpoint │ └── model.pth ├── config.py ├── data │ ├── predict.csv │ └── train.csv ├── dataset.py ├── evaluator.py ├── figure ├── log ├── main.py ├── model │ └── net_regression.py ├── tool │ ├── generate_data.py │ └── merge_csv.py ├── trainner.py └── util
【文件预览】:
LinearRegression-master
----config.py(2KB)
----evaluator.py(993B)
----Train1.py(3KB)
----model()
--------net_regression.py(489B)
----utils.py(1KB)
----main.py(2KB)
----trainner.py(4KB)
----dataset.py(4KB)
----Result3.py(787B)
----README.md(2KB)
----Prediction2.py(3KB)
----data()
--------predict.csv(87B)
--------train.csv(387B)
----tool()
--------merge_csv.py(306B)
--------generate_data.py(4KB)