我使用kaggle:中的住房数据编写了这段代码
X = df['RM'].values.reshape(-1,1)
y = df['MEDV'].values
from sklearn.linear_model import RANSACRegressor
ransac = RANSACRegressor()
ransac.fit(X,y)
inlier_mask = ransac.inlier_mask_
outlier_mask = np.logical_not(inlier_mask)
np.arange(3,10,1)
line_X = np.arange(3,10,1)
line_y_ransac = ransac.predict(line_X.reshape(1,-1))
我在line_y
:上得到了这个错误
ValueError: matmul: Input operand 1 has a mismatch in its core dimension 0, with gufunc signature (n?,k),(k,m?)->(n?,m?) (size 1 is different from 7)
我发现了问题。我应该将-1, 1
而不是1, -1
传递给line_X.reshape
:
line_X.reshape(-1,1))