Have train.py look in subdirectories

Added class TrainANN() to simplify training code.
Method TrainData.split() splits training data into train and test data