Caso mais alguém tenha achado complicada aquela função separa_dados apresentada na aula segue uma alternativa:
#treino
xtreino_novo = pd.DataFrame(ytreino[:-1])
ytreino_novo = ytreino[1:]
#teste
xteste_novo = pd.DataFrame(yteste[:-1])
yteste_novo = yteste[1:]