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

-4 votes97 views4 answers
Python 3.4 release date
Python
0 vote119 views1 answer
Check if host is reachable python
Python
0 vote103 views1 answer
Applymap(lambda x: x.encode('unicode_escape'). decode('utf-8') if isinstance(x, str) else x)
Python
8 votes100 views2 answers
Python f-string padding
Python
0 vote113 views1 answer
Import one file into another python
Python
0 vote112 views2 answers
How to terminate subprocess.call in python
Python
0 vote133 views1 answer
Bad resolution with df plot
Python
0 vote178 views1 answer
Typecasting python
Python
6 votes87 views4 answers
Normalize column pandas
Python
0 vote100 views1 answer
Gensim wordvector vocabulary list
Python