Solucionado (ver solução)
Solucionado
(ver solução)
2
respostas

codigo pronto yellow brick error

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 ??

2 respostas
solução!

Olá Edu.

Acredito que o problema está em como você fez o split do seu treino e teste.

Quando foi separar os dados de teste e treino colocou uma chave a mais :

Y = dados[['nota_total']]

Mas na verdade o Y deve ficar assim:

Y = dados['nota_total']

Do jeito que estava o Y vai ser um DataFrame invés de um Series e então o seu test_y e train_y vão ter a dimensão incompatível depois de utilizar a função train_test_split.

Espero ter ajudado.

Bons Estudos.

testei aqui, e vi que era isso mesmo, já arrumei e deu certo !

muito obrigado pela resposta !

Quer mergulhar em tecnologia e aprendizagem?

Receba a newsletter que o nosso CEO escreve pessoalmente, com insights do mercado de trabalho, ciência e desenvolvimento de software