text_html_output_1.png
from sklearn.model_selection import train_test_split

import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split
CSV_PATH = '/kaggle/input/iris/Iris.csv'
ID = 'Id'
TARGET = 'Species'
TEST_SIZE = 0.3
VAL_SIZE = 0.3
SEED = 2023

y_trainval.value_counts()

X_train, X_val, y_train, y_val = train_test_split(X_trainval, y_trainval, test_size=VAL_SIZE, random_state=SEED, stratify=y_trainval)
print('train set = ', len(X_train), ' val set', len(X_val))