更新 function.py

2024-11-02 21:58:28 +08:00
parent d66e4cd870
commit 1437fce359
1 changed files with 2 additions and 2 deletions
@@ -37,7 +37,7 @@ def load_data(filepath):
            continue
        y.append(1 if data[22] in ['2', '3'] else 0)  # 假设标签是 '2' 或 '3' 为正类
        data_25 = data[24]
-        data = [float(x) if x != '?' else 0 for x in data[:24] + data_25.split(',')]  # 处理缺失值
+        data = [0 if x == '?' else float(x) for x in data[:24] + data_25.split(',')]  # 处理缺失值
        data_list.append(data)
    X = np.array(data_list)
    return X, np.array(y)
@@ -60,5 +60,5 @@ def main():
    accuracy_test = np.mean(y_pred_test == y_test)
    print(f"Test Accuracy: {accuracy_test:.2f}")
-if __name__.split("__main__"):
+if __name__ == "__main__":
    main()