Python Numpy: операнды не могут передаваться вместе с фигурами

Я получаю эту ошибку "операнды не могут быть переданы вместе с фигурами" для этого кода

import numpy as np
from sklearn.datasets import load_boston
from sklearn.linear_model import LinearRegression
beantown = load_boston()
x=beantown.data
y=beantown.target
model = LinearRegression()
model = model.fit(x,y)

def mse(truth, predictions): 
    return ((truth - predictions) ** 2).mean(None)
print model.score(x,y)
print mse(x,y)

Ошибка в строке печати mse(x,y)

Ошибка - ValueError:

operands could not be broadcast together with shapes (506,13) (506,)

1 ответ

Решение

Переформуйте y:

from sklearn.datasets import load_boston
from sklearn.linear_model import LinearRegression
beantown = load_boston()
x = beantown.data
y = beantown.target
y = y.reshape(y.size, 1)
model = LinearRegression()
model = model.fit(x, y)


def mse(truth, predictions):
    return ((truth - predictions) ** 2).mean(None)
print model.score(x, y)
print mse(x, y)
Другие вопросы по тегам