Adicionando particionamento do conjunto de dados para treinamento e teste. Modificações na DataSource