【发布时间】:2021-02-07 18:48:57
【问题描述】:
我是 PyTorch 的新手。谁能向我解释为什么我们应该在执行回归之前将数据帧转换为火炬张量? (如您所见,我已经拆分了数据集)。我还需要对这段代码进行解释:
x_train_tensor = torch.tensor(x_train.values.reshape(-1, 1), dtype = torch.float)
y_train_tensor = torch.tensor(y_train.values.reshape(-1, 1), dtype = torch.float)
x_test_tensor = torch.tensor(x_test.values.reshape(-1, 1), dtype = torch.float)
y_test_tensor = torch.tensor(y_test.values.reshape(-1, 1), dtype = torch.float)
谢谢!
【问题讨论】:
标签: python pytorch tensor torch