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

7 votes276 views4 answers
Pandas.get_dummies
Python
0 vote264 views1 answer
How to use import command in python
Python
0 vote251 views1 answer
Python emoji convert
Python
1 vote256 views1 answer
Squared sum of all elements in list python
Python
1 vote326 views5 answers
Filter function in python
Python
22 votes262 views3 answers
How to sort a list descending python
Python
0 vote329 views1 answer
Django insert bulk data
Python Django
3 votes270 views1 answer
Pandas show top 10 rows
Python
0 vote264 views2 answers
Pyinstaller pymssql
Python
1 vote246 views1 answer
Plt.plot width line
Python