Breaking News: Grepper is joining You.com. Read the official announcement!

how to split a dataframe into train and test

Add Answer

Worried Warbler answered on August 2, 2020 Popularity 10/10 Helpfulness 3/10

answer how to split a dataframe into train and test

related pandas split train test

related pandas split dataframe to train and test

related pandas split train test

related train-test split code in pandas

related python train test split in pandas

how to split a dataframe into train and test

Comment

# Dataframe splitting helper function
def SplitDataframe(df, y_column, test_size=3):
    train_count = int(round(test_size*10/len(df)*100))
    
    train_ds = df[train_count:]
    test_ds = df[:train_count]
    
    train_ds_X = train_ds.drop([y_column], axis=1)
    train_ds_y = train_ds[y_column]
    
    test_ds_X = test_ds.drop([y_column], axis=1)
    test_ds_y = test_ds[y_column]
    
    return (train_ds_X, train_ds_y), (test_ds_X, test_ds_y)

Popularity 10/10 Helpfulness 3/10 Language python

Source: Grepper

Tags: dataframe python

Link to this answer
Share Copy Link

Contributed on Mar 11 2022

Worried Warbler

0 Answers Avg Quality 2/10

Closely Related Answers

pandas split train test

Comment

from sklearn.model_selection import train_test_split


y = df.pop('output')
X = df

X_train,X_test,y_train,y_test = train_test_split(X.index,y,test_size=0.2)
X.iloc[X_train] # return dataframe train

Popularity 10/10 Helpfulness 10/10 Language python

Source: stackoverflow.com

Tags: pandas pa

Link to this answer
Share Copy Link

Contributed on Dec 24 2020

Courageous Cod

0 Answers Avg Quality 2/10

pandas split dataframe to train and test

Comment

train=df.sample(frac=0.8,random_state=200) #random state is a seed value
test=df.drop(train.index)

Popularity 10/10 Helpfulness 8/10 Language python

Source: stackoverflow.com

Tags: dataframe data

Link to this answer
Share Copy Link

Contributed on Aug 02 2020

Courageous Chamois

0 Answers Avg Quality 2/10

pandas split train test

Comment

from sklearn.model_selection import train_test_split

train, test = train_test_split(df, test_size=0.2)

Popularity 10/10 Helpfulness 7/10 Language python

Source: stackoverflow.com

Tags: pandas pa

Link to this answer
Share Copy Link

Contributed on Dec 24 2020

Courageous Cod

0 Answers Avg Quality 2/10

train-test split code in pandas

Comment

df_permutated = df.sample(frac=1)

train_size = 0.8
train_end = int(len(df_permutated)*train_size)

df_train = df_permutated[:train_end]
df_test = df_permutated[train_end:]

Popularity 10/10 Helpfulness 4/10 Language python

Source: Grepper

Tags: pandas pa

Link to this answer
Share Copy Link

Contributed on Nov 18 2020

Coder_Fox

0 Answers Avg Quality 2/10

python train test split in pandas

Comment

df_test=df.drop(df_train.index)

Popularity 9/10 Helpfulness 2/10 Language python

Source: pub.towardsai.net

Tags: pandas pa

Link to this answer
Share Copy Link

Contributed on Dec 07 2022

Caio Henrique

0 Answers Avg Quality 2/10

python train test split in pandas

Comment

-1

df_train = df.sample(frac=0.8, random_state=1)

Popularity 8/10 Helpfulness 1/10 Language python

Source: pub.towardsai.net

Tags: pandas pa

Link to this answer
Share Copy Link

Contributed on Dec 07 2022

Caio Henrique

0 Answers Avg Quality 2/10

how to split a dataframe into train and test

Contents

More Related Answers

how to split a dataframe into train and test

Closely Related Answers

pandas split train test

pandas split dataframe to train and test

pandas split train test

train-test split code in pandas

python train test split in pandas

python train test split in pandas

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.