def _get_train_data_loader(batch_size, training_dir):
    print('Get train data loader.')
    train_data = pd.read_csv(os.path.join(training_dir, 'train.csv'), header=None, names=None)
    train_y = torch.from_numpy(train_data[[0]].values).float().squeeze()
    train_x = torch.from_numpy(train_data.drop([0], axis=1).values).float()
    train_ds = torch.utils.data.TensorDataset(train_x, train_y)
    return torch.utils.data.DataLoader(train_ds, batch_size=batch_size)
def _get_train_data_loader(batch_size, training_dir):
    print('Get train data loader.')
    train_data = pd.read_csv(os.path.join(training_dir, 'train.csv'), header=None, names=None)
    train_y = torch.from_numpy(train_data[[0]].values).float().squeeze()
    train_x = torch.from_numpy(train_data.drop([0], axis=1).values).float()
    train_ds = torch.utils.data.TensorDataset(train_x, train_y)
    return torch.utils.data.DataLoader(train_ds, batch_size=batch_size)