from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
# 1. Исправлена закрывающая скобка в data[features]
X, y = data[features], data["is_male"]
# 2. Исправлено имя параметра: test_size (через нижнее подчеркивание)
X_tr, X_te, y_tr, y_te = train_test_split(
X, y, test_size=0.3, random_state=0, stratify=y
)
# 3. Вычисляем медианы на обучающей выборке
medians = X_tr.median()
# 4. Заполняем пропуски вычисленными медианами
X_tr = X_tr.fillna(medians)
X_te = X_te.fillna(medians)
# 5. Инициализируем и обучаем модель
model = LogisticRegression(max_iter=1000)
model.fit(X_tr, y_tr)
# 6. Исправлен знак равенства при присваивании переменной score
score = model.score(X_te, y_te)