Quando tento plotar o grafico com yellowbrick, retorna o erro "illegal value in 4-th argument of internal None". Como eu poderia resolver esse problema? Segue o codigo:
from yellowbrick.regressor import ResidualsPlot
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()
visualizer = ResidualsPlot(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)