更新 function.py

function.py

@@ -10,7 +10,7 @@ id = "202210423210"
 name = "陈宇杰"
 
 """
-函数名称: load_data
+函数名称: load_dataset
 功能: 加载训练数据集和测试数据集。
 参数:
 - directory (str): 包含手写数字图像的文件夹路径。
@@ -18,7 +18,7 @@ name = "陈宇杰"
 - dataSet (numpy array): 训练数据集或测试数据集，shape 为 (m, 1024)。
 - labels (numpy array): 训练集或测试集对应的标签，shape 为 (m,)。
 """
-def load_data(directory):
+def load_dataset(directory):
     file_list = os.listdir(directory)
     data_matrix = []
     labels = []
@@ -100,8 +100,8 @@ def normalize_data(dataSet):
 """
 def main():
     # 加载训练集和测试集
-    training_data, training_labels = load_data('trainingDigits')
-    test_data, test_labels = load_data('testDigits')
+    training_data, training_labels = load_dataset('trainingDigits')
+    test_data, test_labels = load_dataset('testDigits')
     
     # 归一化数据
     training_data = normalize_data(training_data)