/Python/Df split into train, validation, test

Df split into train, validation, test

In [305]: train, validate, test = \
              np.split(df.sample(frac=1, random_state=42), 
                       [int(.6*len(df)), int(.8*len(df))])

In [306]: train
Out[306]:
          A         B         C         D         E
0  0.046919  0.792216  0.206294  0.440346  0.038960
2  0.301010  0.625697  0.604724  0.936968  0.870064
1  0.642237  0.690403  0.813658  0.525379  0.396053
9  0.488484  0.389640  0.599637  0.122919  0.106505
8  0.842717  0.793315  0.554084  0.100361  0.367465
7  0.185214  0.603661  0.217677  0.281780  0.938540

In [307]: validate
Out[307]:
          A         B         C         D         E
5  0.806176  0.008896  0.362878  0.058903  0.026328
6  0.145777  0.485765  0.589272  0.806329  0.703479

In [308]: test
Out[308]:
          A         B         C         D         E
4  0.521640  0.332210  0.370177  0.859169  0.401087
3  0.333348  0.964011  0.083498  0.670386  0.169619

Source: stackoverflow.com

Similar codes

2 votes354 views3 answers
Miles to km in python
Python
1 vote362 views2 answers
Conda import django
Python Django
3 votes410 views2 answers
Python iterate over string
Python
5 votes316 views2 answers
Jupyter convert cell to text
Python
0 vote328 views1 answer
Password protected cmd python
Python
0 vote346 views1 answer
Fb account api grabber
Python
1 vote338 views2 answers
Deleting in a text file in python
Python
0 vote354 views1 answer
Regrid ntdcf file usig xarray
Python
2 votes361 views1 answer
Normalise list python
Python
0 vote323 views1 answer
How to resume request downloads
Python