diff --git a/function.py b/function.py
index 9c2a024..92d47bd 100644
--- a/function.py
+++ b/function.py
@@ -1,5 +1,6 @@
 # function.py
 import numpy as np
+from sklearn.preprocessing import StandardScaler
 
 """
 挑战说明: