pytorch的Linear, 简单的线性回归模型

import random

import torch
import torch.nn as nn

input_dimension = 3
output_dimension = 1
# shujuliang: 数据量的个数
shujuliang = 2
# 定义一个简单的线性回归模型
class LinearRegressionModel(nn.Module):
    def __init__(self):
        super(LinearRegressionModel, self).__init__()
        self.linear = nn.Linear(in_features=input_dimension, out_features=output_dimension, bias=True , dtype=torch.double)  # 输入维度为1，输出维度为1

    def forward(self, x):
        return self.linear(x)

# 实例化模型
model = LinearRegressionModel()


linear_layer = model.linear

random_list = [[random.randint(0, 1) for _ in range(input_dimension)] for _ in range(shujuliang)]

shuru = torch.tensor(random_list, dtype=torch.double)
print(f"shuru:{shuru}")
shuchu = model.forward(shuru)
print(f'shuchu:{shuchu}')
right_shuchu = torch.randint(0, 2, (shujuliang, 1))
print('right_shuchu:', right_shuchu)
# weights维度是input_dimension*output_dimension
print(f'Weights: {linear_layer.weight.data}')
# bias维度是(output_dimension,)
print(f'Bias: {linear_layer.bias.data}')

# 定义损失函数和优化器
criterion = nn.MSELoss()
loss = criterion(shuchu, right_shuchu)
print(f"loss: {loss}")