python - 将两个参数传递给 skicit model.predict()

Question

我做这个是为了练习线性回归算法：

from sklearn.linear_model import LinearRegression
model = LinearRegression()

X = housing[['lotsize',"bedrooms"]]
y = housing[['price']]

model.fit(X, y)

我被困在我应该调用我的结果的下一部分，我知道如何使用 1 个参数来做到这一点，例如：model.predict([[400]])如果我只使用lotsize

model.predict如果我给它两个论点，我该怎么写？（lotsize 和卧室，取值在 0 到 3 之间）

我确信这很简单，但我很难找到正确的语法......

标签： pythonscikit-learn

Answer 1

尝试类似：

model.predict([[400, 1]]) #[[x1, x2]]