Não consegui entender bem o motivo de atribuirmos ao parâmetro stratify a variável y. Alguém poderia explicar com mais detalhes/exemplos?
Você está vendo a versão anterior da nova experiência da Alura que estamos preparando para você. Em breve, ela ganha uma identidade visual novinha totalmente pensada em potencializar seus estudos!
Não consegui entender bem o motivo de atribuirmos ao parâmetro stratify a variável y. Alguém poderia explicar com mais detalhes/exemplos?
O Stratify diz que você quer a mesma proporcao no seu conjunto de treino e testes da variavel escolhida.
Por exemplo:
Var y = [0, 0, 0, 0, 0, 0, 1, 1, 1, 1] (60% zero e 40% um)
Vamos supor que voce queira treinar com 5 amostras, então seu output deve ser: Var y = [0, 0, 0, 1, 1] (60% zero e 40% um)
Lembrando que que ele vai escolher o X baseado nisso.