Torch split classes stratified

X = np.random.randn(1000, 2)
y = np.random.randint(0, 10, size=1000)

X_train, X_val, y_train, y_val = train_test_split(X, y, test_size=0.1, stratify=y)
np.unique(y_train, return_counts=True)
np.unique(y_val, return_counts=True)

train_dataset = Dataset(X_train, y_train, ...)
train_loader = DataLoader(train_dataset, ...)

Source: discuss.pytorch.org

Similar codes

0 vote329 views1 answer
How to calculate the variance of all columns in python
Python
0 vote323 views1 answer
Nearest neighbors imputation
Python
4 votes330 views1 answer
Get prime number python
Python
0 vote387 views1 answer
How to get table schema sql pyodbc
Python
0 vote309 views1 answer
How to create enter pressed for qlineedit in pyqt5
Python
0 vote322 views1 answer
Error tokenizing data. C error: Calling read(nbytes) on source failed. Try engine='python'.
Python
19 votes305 views3 answers
Convert a list ['Pythons syntax is easy to learn', 'Pythons syntax is very clear'] to a string using ‘&’.
Python
0 vote362 views1 answer
For loop in django template css
Python Django
0 vote347 views1 answer
Access key through value python
Python
10 votes329 views2 answers
Print key of dictionary python
Python