cy/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 40s
Details

Browse Source
This commit is contained in:
cy
2024-10-20 14:25:33 +08:00
parent b62ce0b32b
commit dbf4359cec
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
function.py
+8 -3
View File
@@ -1,6 +1,5 @@
import numpy as np import numpy as np
import os import os
from sklearn.preprocessing import StandardScaler
""" """
@@ -103,8 +102,14 @@ def load_dataset(directory):
labels.append(label) labels.append(label)
data_set.append(img_to_vector(os.path.join(root, file))) data_set.append(img_to_vector(os.path.join(root, file)))
data_set = np.array(data_set).reshape(len(data_set), - 1) data_set = np.array(data_set).reshape(len(data_set), - 1)
scaler = StandardScaler()
data_set = scaler.fit_transform(data_set) # 找到数据集中的最大值和最小值
data_min = np.min(data_set)
data_max = np.max(data_set)
# 归一化数据
data_set = (data_set - data_min) / (data_max - data_min)
return data_set, np.array(labels) return data_set, np.array(labels)