Pandas to tensor torch

train_target = torch.tensor(train['Target'].values.astype(np.float32))
train = torch.tensor(train.drop('Target', axis = 1).values.astype(np.float32)) 
train_tensor = data_utils.TensorDataset(train, train_target) 
train_loader = data_utils.DataLoader(dataset = train_tensor, batch_size = batch_size, shuffle = True)

Source: stackoverflow.com

Similar codes

1 vote268 views1 answer
Assert len(lex) < self.bucket_specs[-1][1]
Python
3 votes277 views3 answers
How to remove items from list in python
Python
0 vote261 views4 answers
Python call function from string
Python
0 vote332 views2 answers
How to get current page url in django template
Python Django
4 votes292 views4 answers
Binary to decimal in python
Python
7 votes271 views2 answers
Check package version python
Python
0 vote303 views1 answer
Python how to code discord bot kick members
Python
1 vote291 views1 answer
Who wrote permission to dance
Python
0 vote304 views1 answer
'djdt' is not a registered namespace
Python
-1 vote262 views1 answer
Pytube3
Python