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

1 vote332 views1 answer
Df.select_dtypes
Python
5 votes402 views9 answers
Email regex
Python
0 vote383 views1 answer
Countplot for different classes in a column
Python
15 votes334 views3 answers
Machine learning python
Python
4 votes328 views1 answer
End python in command
Python
0 vote356 views1 answer
Python sum lists element wise
Python
-1 vote337 views1 answer
Sorting numbers in python without sort function
Python
0 vote374 views1 answer
Generate 50 characters long for django
Python
3 votes318 views2 answers
Django save methods
Python
8 votes330 views2 answers
Determinant of a matrix in python
Python