Train,test,dev python

import numpy as np
import pandas as pd

def train_validate_test_split(df, train_percent=.6, validate_percent=.2, seed=None):
    np.random.seed(seed)
    perm = np.random.permutation(df.index)
    m = len(df.index)
    train_end = int(train_percent * m)
    validate_end = int(validate_percent * m) + train_end
    train = df.iloc[perm[:train_end]]
    validate = df.iloc[perm[train_end:validate_end]]
    test = df.iloc[perm[validate_end:]]
    return train, validate, test

Source: stackoverflow.com

Similar codes

6 votes128 views2 answers
How to sort a list randomly in python
Python
29 votes262 views5 answers
Static dirs django
Python Django
3 votes121 views2 answers
Django postgres connection
Python
0 vote122 views1 answer
How to read frame width of video in cv2
Python
0 vote128 views1 answer
Hex string to hex number
Python
4 votes126 views3 answers
Dice simulator python
Python
0 vote126 views2 answers
Select list of columns pandas
Python
0 vote118 views1 answer
Making constructor in python
Python
25 votes125 views2 answers
Python copy to clipboard command
Python
0 vote127 views1 answer
How to store something in python
Python