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

2 votes325 views1 answer
Custom signal godot
Python
0 vote408 views6 answers
Python converter to c
Python
0 vote388 views1 answer
Django import excel file from same directory
Python Django
0 vote333 views1 answer
Python sum of 10 numbers from user input
Python
1 vote352 views1 answer
Panda dataframe read csv change string to float
Python
0 vote342 views1 answer
Python select certain characters from string
Python
0 vote337 views1 answer
How to create correlation heatmap in python
Python
0 vote318 views1 answer
Python types generator
Python
0 vote318 views1 answer
Speak by a discord bot in python
Python
0 vote339 views1 answer
Truncate string python
Python