Template
更新 function.py
This commit is contained in:
+2
-2
@@ -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()
|
||||
Reference in New Issue
Block a user