train,test,dev python

Victorious Vendace answered on July 17, 2020 Popularity 6/10 Helpfulness 6/10

answer train,test,dev python

train,test,dev python

Comment

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

Popularity 6/10 Helpfulness 6/10 Language python

Source: stackoverflow.com

Tags: python

Link to this answer
Share Copy Link