Oi, Marcia. Tudo bem?
Possivelmente o seu y_teste possui tamanho não esperado para que o método seja aplicado. Para verifiar esse tamanho, entre com o comando:
print(treino_y.shape)
Tenda ver se isso te ajuda. Se não, por favor, poste o código aqui pra eu tentar dar uma olhada.
Atenciosamente, Luís