尝试训练数据时使用 pytorch 的大小不匹配答案

【问题标题】：Size Mismatch using pytorch when trying to train data尝试训练数据时使用 pytorch 的大小不匹配
【发布时间】：2020-02-03 14:38:39
【问题描述】：

我对 pytorch 真的很陌生，只是想用我自己的数据集来做一个简单的线性回归模型。我也只使用数字值作为输入。

我已从 CSV 导入数据

dataset = pd.read_csv('mlb_games_overview.csv')

我已经把数据分成了四部分 X_train, X_test, y_train, y_test

X = dataset.drop(['date', 'team', 'runs', 'win'], 1)
y = dataset['win']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=True)

我已将数据转换为 pytorch 张量

X_train = torch.from_numpy(np.array(X_train))
X_test = torch.from_numpy(np.array(X_test))
y_train = torch.from_numpy(np.array(y_train))
y_test = torch.from_numpy(np.array(y_test))

我创建了一个线性回归模型

class LinearRegressionModel(torch.nn.Module):
    def __init__(self):
        super(LinearRegressionModel, self).__init__()
        self.linear = torch.nn.Linear(1, 1)
    def forward(self, x):
        y_pred = self.linear(x)
        return y_pred

我已经初始化了优化器和损失函数

criterion = torch.nn.MSELoss(reduction='sum')
optimizer = torch.optim.SGD(model.parameters(), lr=0.01)

现在当我开始训练数据时，我得到运行时错误不匹配

EPOCHS = 500
for epoch in range(EPOCHS):
    pred_y = model(X_train) # RUNTIME ERROR HERE
    loss = criterion(pred_y, y_train)
    optimizer.zero_grad() # zero out gradients to update parameters correctly
    loss.backward() # backpropagation
    optimizer.step() # update weights
    print('epoch {}, loss {}'. format(epoch, loss.data[0]))

错误日志：

RuntimeError                              Traceback (most recent call last)
<ipython-input-40-c0474231d515> in <module>
  1 EPOCHS = 500
  2 for epoch in range(EPOCHS):
----> 3     pred_y = model(X_train)
  4     loss = criterion(pred_y, y_train)
  5     optimizer.zero_grad() # zero out gradients to update parameters correctly
RuntimeError: size mismatch, m1: [3540 x 8], m2: [1 x 1] at 
C:\w\1\s\windows\pytorch\aten\src\TH/generic/THTensorMath.cpp:752

【问题讨论】：

标签： machine-learning pytorch linear-regression tensor

【解决方案1】：

在您的线性回归模型中，您有：

self.linear = torch.nn.Linear(1, 1)

但是您的训练数据 (X_train) 形状是 3540 x 8，这意味着您有 8 个特征代表每个输入示例。所以，你应该如下定义线性层。

self.linear = torch.nn.Linear(8, 1)

PyTorch 中的linear layer 具有参数W 和b。如果将in_features 设置为8 并将out_features 设置为1，则W 矩阵的形状将为1 x 8，b 向量的长度将为1。

由于您的训练数据形状为3540 x 8，您可以执行以下操作。

linear_out = X_train W_T + b

我希望它能澄清你的困惑。

【讨论】：