Erro no comando scatterplot | Machine Learning: classificação com SKLearn

ValueError Traceback (most recent call last) <ipython-input-15-7f95ef61d3e4> in <module>() ----> 1 sms.scatterplot(x="horas_esperadas", y="preco", hue=teste_y, data=teste_x) 11 frames /usr/local/lib/python3.6/dist-packages/pandas/core/internals/managers.py in _stack_arrays(tuples, dtype) 1874 stacked = np.empty(shape, dtype=dtype) 1875 for i, arr in enumerate(arrays): -> 1876 stacked[i] = _asarray_compat(arr) 1877 1878 return stacked, placement ValueError: could not broadcast input array from shape (540) into shape (1)

Olá Adriana.

Acredito que uma de suas variáveis ou teste_y ou teste_x está com o valor incorreto, poderia verifica-las para nos conseguirmos identificar uma solução.

Poderia colocar esses prints:

import seaborn as sns
print(teste_x.shape)
print(teste_y.shape)
sns.scatterplot(x="horas_esperadas", y="preco", hue=teste_y, data=teste_x)

E verificar se o resultado é igual a esse:

(540, 2)
(540,)

Caso não esteja igual, poderia verificar se suas instruções que precedem estão corretas:

Primeiro quando atribuímos os valores de x e y:

x = dados[['horas_esperadas', 'preco']]
y = dados['finalizado']

Depois quando separamos os valores e colocamos nas variaveis que podem está com problema teste_y e teste_x.

treino_x, teste_x, treino_y, teste_y = train_test_split(x, y, test_size = 0.25)

Para o funcionamento correto as únicas atribuições nessa variáveis devem ser essas até chegar nessa instrução.

import seaborn as sns
sns.scatterplot(x="horas_esperadas", y="preco", hue=teste_y, data=teste_x)

Caso isso não resolva seu problema, compartilha comigo o seu notebook do Colab, dessa maneira consigo olhar o código com você e encontrar o problema.

Bons Estudos.