Question

我正在研究机器学习项目我在python中做了一个多元线性回归模型，这是我的代码

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
from sklearn.linear_model import LinearRegression

data = pd.read_csv("train.csv", delimiter=",", header=0)
x = data['Col1'][:, np.newaxis]
y = data['Expected']
reg = LinearRegression()
reg.fit(x, y)

我的train.csv文件包含3列Col1，Col2，预期，因此x将包含输入“Col1”和“Col2”记录，而y将是putput“预期”记录。我只能在x中放入一个输入，即Col1，但我不能将Col2放在x中。

如何在x中保存两列Col1和Col2的值，以便在以后的线性回归中拟合？

>>>> print(data.head())
          Col1     Col2  Expected
      0   7.645   5.2119      46.0
      1   7.079   3.4145      28.7
      2  91.900  24.0000      50.0
      3   5.875   1.1296      50.0
      4   6.153   3.2797      29.6

Answer 1

你需要的只是

x = data[['Col1', 'Col2']]

多元线性回归输入拟合

1 个答案: