codigo para Dividir el conjunto entre datos de entrenamiento y datos de prueba, los datos de prueba serán el 15% del dataset

def train_test_split(X, y, test_size = 0.15):
    X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = test_size, random_state = 54)
    return X_train, X_test, y_train, y_test