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

0 vote103 views1 answer
String in netcdf file python
Python
-1 vote106 views2 answers
Append multiple elements
Python
6 votes103 views1 answer
Write a Python program to append text to a file and display the text.
Python
0 vote90 views1 answer
Win64pyinstaller
Python
13 votes112 views4 answers
Pandas take first n rows
Python
0 vote98 views1 answer
ImportError: cannot import name 'clock' from 'time' (unknown location)
Python
9 votes109 views4 answers
Change column order pandas
Python
79 votes106 views6 answers
String reverse in python
Python
2 votes100 views1 answer
Tkinter execute function on enter
Python
0 vote102 views1 answer
Pandas loop over chunk of rows
Python