diff --git a/function.py b/function.py index 471bad3..3b1d494 100644 --- a/function.py +++ b/function.py @@ -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() \ No newline at end of file