add training_dataset and test_dataset