input train and test data directory