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 vote127 views3 answers
How to compare the two key from constant value to list of string in python
Python
0 vote179 views2 answers
Python suppress exponential notation
Python
75 votes170 views3 answers
How to get a random element from an array in python
Python
1 vote195 views2 answers
Np one hot encoding
Python
15 votes181 views3 answers
Not in pandas condition
Python
-1 vote196 views1 answer
How to save frames in form of video in opencv python
Python
-2 votes175 views1 answer
Whole loop in python
Python
2 votes173 views3 answers
Torch.stack example
Python
2 votes197 views1 answer
How to return total elements in database django
Python Django
33 votes195 views6 answers
Set in python
Python