generated from LufsX/Experiment-LogReg
This commit is contained in:
+2
-2
@@ -86,7 +86,7 @@ def load_data(filepath):
|
||||
values = [float(v) for v in values]
|
||||
|
||||
# 合并特征23的取值
|
||||
if values[22] == 2 or values[22] == 3: # 特征23(索引22)
|
||||
if values[22] == 2 or values[22] == 3: # 特征23(索引22)
|
||||
labels.append(0) # 死亡
|
||||
elif values[22] == 1:
|
||||
labels.append(1) # 生存
|
||||
@@ -143,4 +143,4 @@ if __name__ == "__main__":
|
||||
|
||||
# 测试单个数据
|
||||
y_pred = predict_logreg(X_test[11], weights)
|
||||
print(f"预测结果: {y_pred}, 实际结果: {y_test[11]}")
|
||||
print(f"预测结果: {y_pred}, 实际结果: {y_test[11]}")
|
||||
Reference in New Issue
Block a user