Fixes #169. Changed .get_train_set to exclude NA values before choosing training set.