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?
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.