本文整理汇总了Python中sklearn.neighbors.ball_tree.BallTree.query_radius方法的典型用法代码示例。如果您正苦于以下问题:Python BallTree.query_radius方法的具体用法?Python BallTree.query_radius怎么用?Python BallTree.query_radius使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类sklearn.neighbors.ball_tree.BallTree
的用法示例。
在下文中一共展示了BallTree.query_radius方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_ball_tree_query_radius
# 需要导入模块: from sklearn.neighbors.ball_tree import BallTree [as 别名]
# 或者: from sklearn.neighbors.ball_tree.BallTree import query_radius [as 别名]
def test_ball_tree_query_radius(n_samples=100, n_features=10):
np.random.seed(0)
X = 2 * np.random.random(size=(n_samples, n_features)) - 1
query_pt = np.zeros(n_features, dtype=float)
eps = 1E-15 # roundoff error can cause test to fail
bt = BallTree(X, leaf_size=5)
rad = np.sqrt(((X - query_pt) ** 2).sum(1))
for r in np.linspace(rad[0], rad[-1], 100):
ind = bt.query_radius(query_pt, r + eps)[0]
i = np.where(rad <= r + eps)[0]
ind.sort()
i.sort()
assert_array_almost_equal(i, ind)
示例2: test_ball_tree_query_radius_distance
# 需要导入模块: from sklearn.neighbors.ball_tree import BallTree [as 别名]
# 或者: from sklearn.neighbors.ball_tree.BallTree import query_radius [as 别名]
def test_ball_tree_query_radius_distance(n_samples=100, n_features=10):
np.random.seed(0)
X = 2 * np.random.random(size=(n_samples, n_features)) - 1
query_pt = np.zeros(n_features, dtype=float)
eps = 1E-15 # roundoff error can cause test to fail
bt = BallTree(X, leaf_size=5)
rad = np.sqrt(((X - query_pt) ** 2).sum(1))
for r in np.linspace(rad[0], rad[-1], 100):
ind, dist = bt.query_radius(query_pt, r + eps, return_distance=True)
ind = ind[0]
dist = dist[0]
d = np.sqrt(((query_pt - X[ind]) ** 2).sum(1))
assert_array_almost_equal(d, dist)