Answers for "how to split data to train and test with sklearn"

30

train test split sklearn

from sklearn.model_selection import train_test_split

X = df.drop(['target'],axis=1).values   # independant features
y = df['target'].values					# dependant variable

# Choose your test size to split between training and testing sets:
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25, random_state=42)
Posted by: Guest on November-08-2020
2

train dev test split sklearn

train, validate, test = np.split(df.sample(frac=1), [int(.6*len(df)), int(.8*len(df))])
Posted by: Guest on August-17-2020

Code answers related to "how to split data to train and test with sklearn"

Python Answers by Framework

Browse Popular Code Answers by Language