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]
|
values = [float(v) for v in values]
|
||||||
|
|
||||||
# 合并特征23的取值
|
# 合并特征23的取值
|
||||||
if values[22] == 2 or values[22] == 3: # 特征23(索引22)
|
if values[22] == 2 or values[22] == 3: # 特征23(索引22)
|
||||||
labels.append(0) # 死亡
|
labels.append(0) # 死亡
|
||||||
elif values[22] == 1:
|
elif values[22] == 1:
|
||||||
labels.append(1) # 生存
|
labels.append(1) # 生存
|
||||||
@@ -143,4 +143,4 @@ if __name__ == "__main__":
|
|||||||
|
|
||||||
# 测试单个数据
|
# 测试单个数据
|
||||||
y_pred = predict_logreg(X_test[11], weights)
|
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