import warnings
from sklearn.datasets import load_iris
from sklearn.exceptions import ConvergenceWarning
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression

# Ignore the convergence warning
warnings.filterwarnings("ignore", category=ConvergenceWarning)

data = load_iris()
X_train, X_test, y_train, y_test = train_test_split(data.data, data.target, random_state=42)

lr_model = LogisticRegression()
lr_model.fit(X_train, y_train)