cool/Experiment-kNN
1
0
Fork 0
generated from LufsX/Experiment-kNN
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新 function.py
Gitea Actions Experiment Test / Experiment-Actions (push) Successful in 1m6s
Details

Browse Source
This commit is contained in:
cool
2024-10-17 09:02:05 +08:00
parent 9ad24a9ee4
commit ff7c19e90c
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
function.py
+4 -4
View File
@@ -10,7 +10,7 @@ id = "202210423210"
name = "陈宇杰" name = "陈宇杰"
""" """
函数名称: load_data 函数名称: load_dataset
功能: 加载训练数据集和测试数据集。 功能: 加载训练数据集和测试数据集。
参数: 参数:
- directory (str): 包含手写数字图像的文件夹路径。 - directory (str): 包含手写数字图像的文件夹路径。
@@ -18,7 +18,7 @@ name = "陈宇杰"
- dataSet (numpy array): 训练数据集或测试数据集，shape 为 (m, 1024)。 - dataSet (numpy array): 训练数据集或测试数据集，shape 为 (m, 1024)。
- labels (numpy array): 训练集或测试集对应的标签，shape 为 (m,)。 - labels (numpy array): 训练集或测试集对应的标签，shape 为 (m,)。
""" """
def load_data(directory): def load_dataset(directory):
file_list = os.listdir(directory) file_list = os.listdir(directory)
data_matrix = [] data_matrix = []
labels = [] labels = []
@@ -100,8 +100,8 @@ def normalize_data(dataSet):
""" """
def main(): def main():
# 加载训练集和测试集 # 加载训练集和测试集
training_data, training_labels = load_data('trainingDigits') training_data, training_labels = load_dataset('trainingDigits')
test_data, test_labels = load_data('testDigits') test_data, test_labels = load_dataset('testDigits')
# 归一化数据 # 归一化数据
training_data = normalize_data(training_data) training_data = normalize_data(training_data)