Estava tentando utilizar o Seaborn com o Pyplot, mas não funcionou a legenda. Então utilizei o seguinte código para ter o gráfico.
resultados.plot(x='max_depth', y=['train', 'test'])
Estava tentando utilizar o Seaborn com o Pyplot, mas não funcionou a legenda. Então utilizei o seguinte código para ter o gráfico.
resultados.plot(x='max_depth', y=['train', 'test'])
Olá Lucas, tudo bem com você?
Peço desculpas pela demora em obter um retorno.
Sua solução alternativa é funcional e poderá inclusive auxiliar outros alunos que passem por situações semelhantes.
Porém, vale ressaltar que no seaborn em conjunto com o pyplot também utilizaremos a função legend para colocar as legendas no gráfico, mas, precisamos nos certificar de adicionar uma label no momento de montar o gráfico (lineplot) ou então, passar os dados da legenda em uma lista.
Exemplo:
import seaborn as sns
import matplotlib.pyplot as plt
x = list(range(0, 11))
y1 = [0, 1, 4, 9, 16, 25, 16, 9, 4, 1, 0]
y2 = [0, 1, 3, 6, 9, 12, 9, 6, 3, 1, 0]
sns.lineplot(x=x, y=y1,label="Série 1")
sns.lineplot(x=x, y=y2,label="Série 2")
# Adicione uma legenda
plt.legend() # ou plt.legend(["Série 1", "Série 2"])
plt.show()
Espero ter ajudado.
Em caso de dúvidas estou à disposição.
Abraços e bons estudos!