Pessoal, Ao inserir o 'teste_y' no 'hue' dentro do scatterplot, recebo o erro: ValueError: Data must be 1-dimensional. Segue o código para análise:
from sklearn.svm import LinearSVC
from sklearn.metrics import accuracy_score
from sklearn.model_selection import train_test_split
import pandas as pd
import seaborn as sns
uri = "https://gist.githubusercontent.com/guilhermesilveira/1b7d5475863c15f484ac495bd70975cf/raw/16aff7a0aee67e7c100a2a48b676a2d2d142f646/projects.csv"
dados = pd.read_csv(uri)
mapa = {'unfinished' : 'nao_concluido' , 'expected_hours' : 'horas_previstas', 'price': 'preco'}
dados = dados.rename(columns = mapa)
# podemos trocar os valores de 1 para 0 e 0 para 1, a fim de facilitar o trabalho:
troca = {
1:0,
0:1
}
dados['Finalizado'] = dados.nao_concluido.map(troca)
SEED = 10
x = dados[['horas_previstas','preco']]
y = dados[['Finalizado']]
treino_x, teste_x, treino_y, teste_y = train_test_split(x,y, random_state = SEED, test_size = 0.25, stratify = y)
model = LinearSVC()
model.fit(treino_x, treino_y)
previsoes = model.predict(teste_x)
acertos = accuracy_score(teste_y, previsoes)
sns.scatterplot(x='horas_previstas', y='preco', hue = teste_y, data=teste_x)
Conseguem me ajudar? Obrigado!