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 vote133 views1 answer
Python cv2 how to update image
Python
0 vote163 views1 answer
Django builtin signals
Python Django
5 votes128 views2 answers
Python get min max value from a dictionary
Python
5 votes141 views4 answers
Python generator function
Python
0 vote119 views1 answer
Python reference to back folder
Python
0 vote151 views1 answer
Free function in python
Python
1 vote131 views1 answer
Qlabel alignment center python
Python
2 votes152 views1 answer
Python turtle background image
Python
73 votes160 views6 answers
Reverse a string in python
Python
0 vote125 views1 answer
Select only object columns pandas
Python