sklearn LinearRegression 报告错误

Question

为什么这在python中不起作用？

    x = []
    y = []
    for ii in range(0,100):
        x.append(ii)
        y.append(ii)
    clf = LinearRegression()
    clf.fit(x, y)
    clf.predict(101)

我收到错误"元组索引超出范围"

Answer 1

为每一行创建一个列表，以便最终得到一个 2D 结构 [[0]， [1]， [2]， ...]：

x = []
y = []
for ii in range(0,100):
    x.append([ii])    <-----
    y.append(ii)
clf = LinearRegression()
clf.fit(x, y)
clf.predict(101)

输出：

数组（[ 101.]）