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 votes177 views2 answers
2d list comprehension python
Python
0 vote184 views2 answers
Wxpython menu callback stackoverflow
Python
0 vote246 views1 answer
Using rstrip() method to remove the newline character from a string
Python
0 vote205 views1 answer
Candle stick with volume plotly
Python
0 vote227 views1 answer
Python pil to greyscale
Python
9 votes227 views4 answers
Pandas not in list
Python
3 votes175 views2 answers
Creating new virtual environment in python
Python
1 vote235 views1 answer
Query with condition django
Python Django
1 vote191 views1 answer
Seaborn jointplot regression
Python
3 votes140 views1 answer
Find all index of element in list python
Python