ao fazer o projeto, cheguei no seguinte código pronto:
código pronto
from yellowbrick.regressor import PredictionError
def visualiza_erros(train_x,train_y,test_x,test_y): visualizer = PredictionError(LinearRegression()) visualizer.fit(train_x, train_y) visualizer.score(test_x, test_y) visualizer.poof()
visualiza_erros(train_x,train_y,test_x,test_y)
que acabou gerando o erro:
"""YellowbrickValueError
...
y must be a (1,) dimensional array not (11067, 1)"""
tentei usar o y.T, mas as dimensões (1, 11067), também não funcionaram
como posso solucionar isso ??
e uma segunda pergunta,
eu posso alterar o código pronto ??