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

158 votes257 views3 answers
Traversing dictionary in python
Python
0 vote252 views1 answer
Python setuptools cythonize except one file
Python
30 votes211 views6 answers
Try and except in python
Python
-3 votes252 views2 answers
Round robin
Python
7 votes252 views1 answer
Replace cell pandas
Python
0 vote317 views1 answer
Using a generic exception block
Python
0 vote231 views1 answer
Matrix diagonal sum leetcode in java
Python
1 vote225 views1 answer
Bgr2gray opencv
Python
1 vote236 views1 answer
Easiest way to position labels in tkinter
Python
0 vote188 views1 answer
Python google docs api how to get doc index
Python