更新 function.py

function.py

@@ -49,7 +49,7 @@ def load_data():
     # 用0填充缺失值  
     data.fillna(0, inplace=True)  
       
-    # 选择特征23作为目标变量，将取值2（死亡）和3（安乐死）合并为一个值0（死亡），并重新编码为二分类问题  
+    # 将取值2（死亡）和3（安乐死）合并为一个值0（死亡），并重新编码为二分类问题  
     data['outcome'] = data['outcome'].apply(lambda x: 0 if x in [2, 3] else 1)  
       
     # 去掉特征3、特征24~28  
@@ -59,9 +59,7 @@ def load_data():
     X = data.drop(columns=['outcome'])  
     y = data['outcome']  
       
-    # 划分训练集和测试集（这里我们模拟划分，实际应使用给定的300个训练样本和68个测试样本）  
+    # 划分训练集和测试集  
-    # 由于没有明确的划分方式，我们假设前300个为训练样本，后68个为测试样本（这通常不是实际做法，仅作为示例）  
-    # 注意：实际应用中应根据数据集提供的信息来准确划分训练集和测试集  
     X_train, X_test = X.iloc[:300], X.iloc[300:]  
     y_train, y_test = y.iloc[:300], y.iloc[300:]  
       
@@ -90,3 +88,4 @@ if __name__ == "__main__":
     y_pred = predict_logreg(model, X_test)  
     accuracy = evaluate_model(y_test, y_pred)  
     print(f"Model Accuracy: {accuracy:.4f}")
+