diff --git a/function.py b/function.py index c2f90ea..a670cbc 100644 --- a/function.py +++ b/function.py @@ -56,10 +56,9 @@ def load_data(filepath): X = data[:, :-1] # 特征数据(假设最后一列是目标变量) y = data[:, 23] # 特征23作为目标变量 - # 转换目标变量为 1 (lived), 2 (died), 3 (euthanized) - y = np.where(y == 2, 0, y) # 2 -> 0 (died -> death) - y = np.where(y == 3, 0, y) # 3 -> 0 (euthanized -> death) - + # 不合并死亡与安乐死,直接使用原始标签 + # 1 = lived, 2 = died, 3 = euthanized + # 将缺失值填充为 0(或其他合适的值) X = np.nan_to_num(X, nan=0) @@ -122,4 +121,4 @@ if __name__ == "__main__": correct += 1 accuracy = correct / len(X_test) * 100 - print(f"测试集准确率: {accuracy}%") + print(f"测试集准确率: {accuracy}%") \ No newline at end of file