batch_size = 16 # Create DataLoader for batch training train_dataset = torch.utils.data.TensorDataset(X_train, y_train) train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=batch_size, shuffle=True) # Create DataLoader for batch training test_dataset = torch.utils.data.TensorDataset(X_test, y_test) test_loader = torch.utils.data.DataLoader(test_dataset, batch_size=batch_size, shuffle=False)