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

-1 vote200 views3 answers
Ceil function in python
Python
16 votes181 views3 answers
How to code music in python
Python
1 vote153 views1 answer
Discord music queue python
Python
1 vote149 views1 answer
Data Compression in python
Python
0 vote152 views1 answer
List files in http directory python
Python
236 votes426 views6 answers
Import random in python
Python
114 votes143 views5 answers
How to use inputs in python
Python
9 votes140 views1 answer
Cannot acess stylesheets in static files
Python
1 vote188 views1 answer
Check string equal with regular expression python
Python
10 votes210 views2 answers
Set axis labels python
Python