generated from LufsX/Experiment-kNN
This commit is contained in:
+5
-4
@@ -98,9 +98,10 @@ def img_to_vector(filename):
|
|||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
# 测试代码
|
# 加载训练数据
|
||||||
train_data, train_labels = load_dataset("trainingDigits")
|
train_data, train_labels = load_dataset("trainingDigits")
|
||||||
|
|
||||||
|
# 测试数据
|
||||||
test_data = (
|
test_data = (
|
||||||
"00000000000000110000000000000000"
|
"00000000000000110000000000000000"
|
||||||
"00000000000011111000000000000000"
|
"00000000000011111000000000000000"
|
||||||
@@ -138,6 +139,6 @@ if __name__ == "__main__":
|
|||||||
|
|
||||||
test_vector = np.array([int(char) for char in test_data]).reshape(1, 1024)
|
test_vector = np.array([int(char) for char in test_data]).reshape(1, 1024)
|
||||||
|
|
||||||
# 使用默认的欧氏距离和不加权的投票机制
|
# 使用曼哈顿距离进行分类
|
||||||
predicted_label = classify(test_vector, train_data, train_labels)
|
predicted_label = classify(test_vector, train_data, train_labels, distance_metric='manhattan')
|
||||||
print(f"预测的标签(普通分类): {predicted_label}")
|
print(f"预测的标签(曼哈顿距离): {predicted_label}")
|
||||||
|
|||||||
Reference in New Issue
Block a user