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 vote131 views1 answer
Chi square test in python
Python
0 vote136 views1 answer
Multiplication of two or more numbers in python
Python
3 votes128 views2 answers
Change column value based on another column pandas
Python
60 votes120 views4 answers
Create exe from python script
Python
1 vote119 views1 answer
Get maximum calue df column
Python
2 votes159 views1 answer
Check if user log in flask
Python Flask
6 votes141 views1 answer
Copy file in python3
Python
0 vote126 views1 answer
How to read linux environment variable in python
Python
2 votes122 views1 answer
How to strip quotation marks in python
Python
0 vote136 views1 answer
Pygame scroll event
Python