Skip to content

Instantly share code, notes, and snippets.

@arunm8489

arunm8489/rnn25

Created August 1, 2020 14:32
Show Gist options
  • Save arunm8489/9e657587ceb776364b83ab588949a820 to your computer and use it in GitHub Desktop.
Save arunm8489/9e657587ceb776364b83ab588949a820 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
rnn25
# create Tensor datasets
train_data = TensorDataset(torch.from_numpy(final_train), torch.from_numpy(y_train.values))
test_data = TensorDataset(torch.from_numpy(final_test), torch.from_numpy(y_test.values))
# dataloaders
batch_size = 500
# make sure to SHUFFLE your data
train_loader = DataLoader(train_data, batch_size=batch_size, shuffle=True)
test_loader = DataLoader(test_data, batch_size=batch_size, shuffle=True )
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment