diff --git a/function.py b/function.py
index 0a6eb28..5768a14 100644
--- a/function.py
+++ b/function.py
@@ -1,5 +1,6 @@
 import numpy as np
 from sklearn.preprocessing import StandardScaler
+from sklearn.linear_model import LogisticRegression
 
 """
 实验说明: