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 vote276 views1 answer
Sns save chart
Python
69 votes294 views6 answers
Python virtualenv
Python
0 vote320 views1 answer
Comtypes python
Python
1 vote284 views1 answer
GdScript onready
Python
1 vote274 views4 answers
Pandas list comprehension
Python
-1 vote288 views1 answer
Django template filter
Python
0 vote284 views1 answer
Get coordinates of an image from a pdf python
Python
2 votes290 views1 answer
Append row to array python
Python
25 votes275 views3 answers
Pyperclip
Python
2 votes364 views2 answers
Python django query
Python Django