更新 function.py
Gitea Actions Experiment Test / Experiment-Actions (push) Successful in 29s

This commit is contained in:
2024-11-03 17:26:32 +08:00
parent 261b7d3bcf
commit b3be0b4923
+3 -3
View File
@@ -58,13 +58,13 @@ def load_data():
X = data.drop(columns=["outcome"]) X = data.drop(columns=["outcome"])
y = data["outcome"] y = data["outcome"]
# 划分训练集和测试集(这里简单按7:3划分,实际应用中应与原数据集划分保持一致) # 划分训练集和测试集(这里简单按7:3划分)
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42) X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
return X_train, X_test, y_train, y_test return X_train, X_test, y_train, y_test
def train_logreg(X_train, y_train): def train_logreg(X_train, y_train):
# 训练对率回归模型 # 训练逻辑回归模型
model = LogisticRegression(max_iter=10000) # 设置足够多的迭代次数以确保收敛 model = LogisticRegression(max_iter=10000) # 设置足够多的迭代次数以确保收敛
model.fit(X_train, y_train) model.fit(X_train, y_train)
return model return model
@@ -74,7 +74,7 @@ def predict_logreg(model, X_test):
y_pred = model.predict(X_test) y_pred = model.predict(X_test)
return y_pred return y_pred
# 示例调用(注:这些调用在自动评测时不会执行,仅供测试使用) # 示例调用
if __name__ == "__main__": if __name__ == "__main__":
X_train, X_test, y_train, y_test = load_data() X_train, X_test, y_train, y_test = load_data()
model = train_logreg(X_train, y_train) model = train_logreg(X_train, y_train)