diff --git a/function.py b/function.py
index b9b2d3b..bd59f45 100644
--- a/function.py
+++ b/function.py
@@ -1,5 +1,6 @@
 import numpy as np
 import os
+from sklearn.preprocessing import StandardScaler
 
 """
 实验说明: