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

5 votes221 views1 answer
If statement in one-line for loop python
Python
0 vote202 views1 answer
Reading a cell from another cell in colab
Python
11 votes226 views3 answers
How to get length of string in python
Python
0 vote225 views1 answer
Class inside class python
Python
2 votes154 views1 answer
Remove , from list python
Python
0 vote178 views1 answer
List tuple dictionary,
Python
2 votes260 views1 answer
Search code ascii python
Python
0 vote206 views2 answers
Pandas series to numpy array
Python
-1 vote176 views2 answers
Dd-mm-yy to yyyy-mm-dd in python
Python
1 vote184 views1 answer
Python how to change back to the later directory
Python