from sklearn.model_selection import train_test_split
from pandas import read_csv
df = read_csv('iris.csv')
X = df[df.columns[1:-1]]
y = df['Species']
X_train, X_test, y_train, y_test = train_testZ_split(X, y)
| Train test split | https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.train_test_split.html |
|---|