diff --git a/function.py b/function.py
index 043fecb..6c764ed 100644
--- a/function.py
+++ b/function.py
@@ -26,6 +26,7 @@ name = "廖运瑞"
 
 
 # 这里写自己的代码实现
+import numpy as np
 from sklearn.preprocessing import MinMaxScaler
 from sklearn.model_selection import train_test_split
 
@@ -139,4 +140,4 @@ if __name__ == "__main__":
 
     # 测试模型
     final_test_accuracy = accuracy(X_test, y_test, final_weights)
-   print(f"最终测试集的准确率: {final_test_accuracy:.4f}, 其他信息: {t[11]}")
\ No newline at end of file
+    print(f"最终测试集的准确率: {final_test_accuracy:.4f}")
\ No newline at end of file