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

0 vote224 views1 answer
Django update replace regex
Python Django
13 votes193 views2 answers
How to know how much lines a file has using python
Python
0 vote217 views1 answer
Visitor IP address django
Python Django
5 votes193 views2 answers
How to empty a text file in python
Python
5 votes217 views4 answers
Pandas categorical to numeric
Python
0 vote185 views1 answer
Compare two excel files using python pandas
Python
8 votes207 views1 answer
Iterative binary search python
Python
0 vote219 views1 answer
Python argparse only allow certain values
Python
2 votes182 views3 answers
If string contains list of letters python
Python
0 vote198 views1 answer
Run exe for python and wait until finish
Python