1
resposta

Dúvidas a respeito do parâmetro Stratify

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?

1 resposta

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.