kkkkkxz/Challenge-kNN
0
0
Fork 0
generated from LufsX/Challenge-kNN
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新 function.py
Gitea Actions Challenge Test / Challenge-Actions (push) Successful in 11s
Details

Browse Source
This commit is contained in:
kkkkkxz
2024-10-08 20:52:43 +08:00
parent 53361ec977
commit 1277773572
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
function.py
+2 -3
View File
@@ -71,12 +71,12 @@ def classify0(inX, dataSet, labels, k=3):
Returns: Returns:
- int: 预测的分类标签。 - int: 预测的分类标签。
""" """
dists = np.sqrt(((dataSet - inX) ** 2).sum(axis=1)) dists = np.sum(np.abs(dataSet - inX), axis=1)
sorted_index = np.argsort(dists) sorted_index = np.argsort(dists)
class_count = {} class_count = {}
for i in range(k): for i in range(k):
vote_label = labels[sorted_index[i]] vote_label = labels[sorted_index[i]]
class_count[vote_label] = class_count.get(vote_label, 0) + 1 / (dists[sorted_index[i]] + 1e-5) # 加权，防止除零 class_count[vote_label] = class_count.get(vote_label, 0) + 1
sorted_class_count = sorted(class_count.items(), key=lambda item: item[1], reverse=True) sorted_class_count = sorted(class_count.items(), key=lambda item: item[1], reverse=True)
if sorted_class_count: if sorted_class_count:
return sorted_class_count[0][0] return sorted_class_count[0][0]
@@ -96,7 +96,6 @@ if __name__ == "__main__":
train_features, train_labels = load_data("datingTrainSet2.tsv") train_features, train_labels = load_data("datingTrainSet2.tsv")
test_features, test_labels = load_data("datingTestSet2.tsv") test_features, test_labels = load_data("datingTestSet2.tsv")
# 归一化特征
train_features = normalize(train_features) train_features = normalize(train_features)
test_features = normalize(test_features) test_features = normalize(test_features)