Na atividade 4 "Curva de Decisão" da aula 3, ele faz um plot dos dados de teste_x, e utiliza como valor da parametro "hue" o teste_y. Copiei no meu colab exatamente do jeito que está no vídeo:
sns.scatterplot(x='horas_esperadas', y='preco', hue=teste_y, data=teste_x)
Mas está dando este erro:
ValueError Traceback (most recent call last) in () ----> 1 sns.scatterplot(x='horas_esperadas', y='preco', hue=treino_y, data=treino_x)
12 frames /usr/local/lib/python3.7/dist-packages/pandas/core/internals/managers.py in verifyintegrity(self) 327 for block in self.blocks: 328 if block.shape[1:] != mgr_shape[1:]: --> 329 raise construction_error(tot_items, block.shape[1:], self.axes) 330 if len(self.items) != tot_items: 331 raise AssertionError(
ValueError: Shape of passed values is (1, 3), indices imply (1682, 3)
Fui até conferir a linha de código que cria teste_y e teste_x, e aparentemente está tudo normal:
from sklearn.model_selection import train_test_split
SEED = 45 #semente de aleatoriedade
treino_x, teste_x, treino_y, teste_y = train_test_split(x, y, test_size=0.22, random_state=SEED, stratify=y)
Por que isto está acontecendo?