我想知道Python库或Tensorflow库中提供了哪些解决方案,用于按属性重要性查找最近邻居。 在KDTree库中,属性具有相同的重要性。
# KDTree
import numpy as np
from sklearn.neighbors import KDTree
# Neighbors and Neighbor
neighbors = np.array([[0, 1, 0, 1],
[1, 2, 1, 2],
[2, 1, 1, 1],
[1, 0, 0, 0],
[1, 2, 0, 2],
[2, 2, 2, 2],
[1, 0, 0, 1]])
neighbor = np.array([[1, 1, 0, 1]])
tree = KDTree(neighbors, leaf_size=2)
distance, ind = tree.query(neighbor, k=7)
为了明确我正在寻找的东西,我开发了以下减半区间算法来计算最近邻居,从左到右依次降低属性的重要性。第一个属性是最重要的,最后一个属性是最不重要的。 https://github.com/itsergiu/Nearest-Neighbor-by-Attribute-Importance