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 vote199 views1 answer
Tkinter how to update optionmenu contents
Python
0 vote181 views1 answer
Cbv uk django
Python
10 votes225 views2 answers
Adding static file and its usage in Django
Python Django
2 votes204 views2 answers
How to store object in file python
Python
0 vote295 views1 answer
Get timestamp with pyrhon
Python
2 votes195 views3 answers
How to set background image in python tkinter
Python
0 vote210 views3 answers
303995
Python
4 votes227 views2 answers
Python regex group
Python
0 vote199 views1 answer
Unable to get local issuer certificate python
Python
19 votes284 views6 answers
Python dictionary to list
Python