Cria uma função separada para realizar o treinamento e predição dos dados.