diff --git a/function.py b/function.py
index cc12a94..8c71266 100644
--- a/function.py
+++ b/function.py
@@ -1,5 +1,8 @@
 # function.py
 import numpy as np
+from sklearn.model_selection import train_test_split
+from sklearn.preprocessing import StandardScaler
+
 
 """
 实验说明: