generated from LufsX/Experiment-kNN
This commit is contained in:
+6
-3
@@ -55,7 +55,7 @@ def classify(input_vector, dataSet, labels, k=3):
|
|||||||
- int: 预测的分类标签。
|
- int: 预测的分类标签。
|
||||||
"""
|
"""
|
||||||
"""
|
"""
|
||||||
欧式距离的 kNN 分类算法。
|
使用欧式距离的 kNN 分类算法。
|
||||||
"""
|
"""
|
||||||
# 计算欧式距离
|
# 计算欧式距离
|
||||||
distances = np.linalg.norm(dataSet - input_vector, axis=1)
|
distances = np.linalg.norm(dataSet - input_vector, axis=1)
|
||||||
@@ -71,7 +71,9 @@ def classify(input_vector, dataSet, labels, k=3):
|
|||||||
|
|
||||||
return most_common
|
return most_common
|
||||||
|
|
||||||
|
|
||||||
# 这里也写自己的代码实现
|
# 这里也写自己的代码实现
|
||||||
|
|
||||||
def load_dataset(directory):
|
def load_dataset(directory):
|
||||||
"""
|
"""
|
||||||
从文件夹中加载数据,若不自行正确实现,则无法通过评测,**不记录成绩**。
|
从文件夹中加载数据,若不自行正确实现,则无法通过评测,**不记录成绩**。
|
||||||
@@ -83,8 +85,8 @@ def load_dataset(directory):
|
|||||||
- np.array: 特征数据。
|
- np.array: 特征数据。
|
||||||
- np.array: 标签数据。
|
- np.array: 标签数据。
|
||||||
"""
|
"""
|
||||||
features = []
|
features = []
|
||||||
labels = []
|
labels = []
|
||||||
|
|
||||||
# 遍历文件夹中的所有文件
|
# 遍历文件夹中的所有文件
|
||||||
for filename in os.listdir(directory):
|
for filename in os.listdir(directory):
|
||||||
@@ -102,6 +104,7 @@ labels = []
|
|||||||
|
|
||||||
return np.array(features), np.array(labels)
|
return np.array(features), np.array(labels)
|
||||||
|
|
||||||
|
|
||||||
# 这里还写自己的代码实现
|
# 这里还写自己的代码实现
|
||||||
def img_to_vector(filename):
|
def img_to_vector(filename):
|
||||||
"""
|
"""
|
||||||
|
|||||||
Reference in New Issue
Block a user