Skip to content

Instantly share code, notes, and snippets.

@amankharwal

amankharwal/car prices Secret

Created September 21, 2020 01:54
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save amankharwal/084933d664ab8fb64e677ca676305d4c to your computer and use it in GitHub Desktop.
Save amankharwal/084933d664ab8fb64e677ca676305d4c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
car prices
inputs = torch.Tensor(inputs_array)
targets = torch.Tensor(targets_array)
dataset = TensorDataset(inputs, targets)
train_ds, val_ds = random_split(dataset, [228, 57])
batch_size = 128
train_loader = DataLoader(train_ds, batch_size, shuffle=True)
val_loader = DataLoader(val_ds, batch_size)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment