基于神经网络的肾脏疾病预测模型

构建一个基于神经网络的肾脏疾病预测模型

1. 数据预处理

• ​加载数据：读取 kidney_disease.csv 文件，加载患者医疗数据。
• ​删除冗余特征：移除与预测目标无关的列（如 al, su 等），保留关键特征（如年龄、血压、血糖等）。
• ​处理缺失值：用 np.nan_to_num 将缺失值（NaN）替换为0，但此方法可能不适用于分类特征（例如“是否有糖尿病”列中0可能代表“否”）。

​2. 特征工程

• ​标签定义：假设数据最后一列（第8列）是分类标签（如 classification），标记患者是否患病（二分类问题）。
• ​分类变量编码：对分类特征（如 dm（糖尿病）、cad（冠心病）、appet（食欲））进行独热编码（One-Hot Encoding），将其转换为数值形式供模型处理。

​3. 数据标准化与分割

• ​标准化：使用 StandardScaler 对数值型特征进行标准化（均值0，方差1），消除量纲差异。
• ​数据分割：按8:2比例划分训练集和测试集，确保模型评估的客观性。

​4. 神经网络建模

• ​模型结构：
  • ​输入层：11个输入节点（对应特征数量）。
  • ​隐藏层：2层全连接层，每层6个神经元，激活函数为ReLU。
  • ​输出层：1个神经元，激活函数为Sigmoid，输出患病概率（0~1）。
• ​训练配置：
  • ​损失函数：二元交叉熵（binary_crossentropy），适用于二分类问题。
  • ​优化器：Adam，自适应调整学习率。
  • ​批次训练：每批次7个样本，共训练20轮（epochs）。

​5. 模型评估

• ​预测与阈值处理：对测试集预测概率大于0.5的样本判定为患病。
• ​性能指标：
  • ​混淆矩阵：计算真阳性、假阳性等分类结果。
  • ​准确率：统计模型正确预测的比例。

# 导入必要库
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.metrics import confusion_matrix, accuracy_score
from keras.models import Sequential
from keras.layers import Dense

# 数据预处理
df = pd.read_csv('kidney_disease.csv')
to_drop = ['al','su','rbc','pc','pcc','ba','bgr','pcv','sod','pot','bu','wc','rc','htn','pe','ane']
df.drop(to_drop, axis=1, inplace=True)
df = df.dropna()  # 删除缺失值

# 分割特征与标签
X = df.drop(['id', 'classification'], axis=1)  # 假设标签列名是'classification'
y = df['classification'].apply(lambda x: 1 if x.lower().strip() == 'ckd' else 0)  # 处理标签格式

# 分类列处理
categorical_cols = ['dm', 'cad', 'appet']
for col in categorical_cols:
    # 清洗字符串数据（统一小写并去除空格）
    X[col] = X[col].astype(str).str.strip().str.lower()
    # 生成哑变量
    dummies = pd.get_dummies(X[col], prefix=col, drop_first=True)
    X = pd.concat([X, dummies], axis=1)
X = X.drop(categorical_cols, axis=1)

# 数据标准化
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=1, stratify=y)  # 添加分层抽样
sc = StandardScaler()
X_train = sc.fit_transform(X_train)
X_test = sc.transform(X_test)

# 模型构建
classifier = Sequential()
classifier.add(Dense(units=6, activation='relu', input_dim=X_train.shape[1], kernel_initializer='he_uniform'))  # 更合适的初始化方法
classifier.add(Dense(units=6, activation='relu', kernel_initializer='he_uniform'))
classifier.add(Dense(units=1, activation='sigmoid', kernel_initializer='he_uniform'))
classifier.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])

# 训练
classifier.fit(X_train, y_train, batch_size=7, epochs=20)

# 评估
y_pred = classifier.predict(X_test) > 0.5
print("Confusion Matrix:\n", confusion_matrix(y_test, y_pred))
print("Accuracy:", accuracy_score(y_test, y_pred))