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 vote158 views1 answer
Ajouter élément liste python
Python
0 vote161 views1 answer
Python numpy find nearest value
Python
0 vote158 views1 answer
Bar break matplotlib
Python
0 vote180 views1 answer
Matplotlib set dpi 300
Python
22 votes167 views2 answers
Reset_index(drop=true)
Python
10 votes167 views1 answer
Print first dictionary keys python
Python
0 vote163 views1 answer
How to scale an array between two values python
Python
23 votes200 views3 answers
If any number python
Python
1 vote183 views2 answers
Make python use python3
Python
3 votes167 views2 answers
Matplotlib log scale y axis base
Python