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

33 votes167 views10 answers
Elon musk
Python
0 vote156 views3 answers
Repeat
Python
1 vote151 views1 answer
How to delete print statement from console pythonn
Python
3 votes142 views1 answer
How to change a string to small letter in python
Python
4 votes169 views1 answer
Python gt index in for cycle
Python
1 vote147 views1 answer
Python requests set header cookie
Python
0 vote186 views1 answer
Pyqt loading screen
Python
8 votes144 views4 answers
Numpy standard deviation
Python
12 votes146 views6 answers
Curl in python
Python
2 votes116 views2 answers
Iterate in pandas dataframe
Python