Precisei adicionar uma conversão para "int" nos valores de "tamanho_de_treino" e "tamanho_de_teste", pois estava dando erro :)
porcentagem_de_treino = 0.90
tamanho_de_treino = int(porcentagem_de_treino * len(Y))
tamanho_de_teste = int(len(Y) - tamanho_de_treino)
Olá, Rodrigo! Tudo tranquilo?
Muito obrigado por compartilhar sua solução.
Atualmente o slice não aceita número float como parâmetro, somente inteiros. Caso seja passado valores float como parâmetro gera o seguinte erro:
TypeError: slice indices must be integers or None or have an __index__ methodLogo faz-se necessário esse casting, que é a transformação de um tipo de objeto em outro (nesse caso, de float para int).
Só uma pequena observação, esse casting: tamanho_de_teste = int(len(Y) - tamanho_de_treino) não é necessário, pois o len(Y) é um inteiro e o tamanho_de_treino você transformou para inteiro na linha de cima.
Caso ainda persista alguma dúvida, estou sempre à disposição.
Espero ter ajudado!
:)