scipy.stats.qmc.

geometric_discrepancy#

scipy.stats.qmc.geometric_discrepancy(sample, method='mindist', metric='euclidean')[源代码]#

基于几何属性的给定样本的离散度。

参数:
samplearray_like (n, d)

计算离散度的样本。

method{“mindist”, “mst”}, 可选

要使用的方法。其中一个 mindist 表示最小距离(默认)或 mst 表示最小生成树。

metricstr 或可调用的对象,可选

要使用的距离度量。有关可用度量和默认值,请参阅以下文档:scipy.spatial.distance.pdist

返回:
discrepancyfloat

离散度(更高的值对应于更大的样本均匀性)。

请参阅

discrepancy

说明

离散度可以作为随机样本质量的简单度量。此度量基于样本中点分布的几何属性,例如任意两点之间的最小距离或最小生成树中的平均边长。

值越高,参数空间的覆盖率越高。请注意,这与 scipy.stats.qmc.discrepancy 不同,其中值越低表示样本的质量越高。

还要注意,在使用此函数比较不同的采样策略时,样本大小必须保持不变。

可以从最小生成树中计算出两个指标:平均边长和边长标准差。使用这两个指标比单独使用任一指标都能提供更好的均匀性图景,其中平均值较高、标准差较低更好(有关简要讨论,请参阅 [1])。此函数当前仅计算平均边长。

参考文献

[1]

Franco J. 等。“最小生成树:一种新的方法来评估计算机实验设计质量。”化学计量学和智能实验室系统,97 (2),第 164-169 页,2009 年。

示例

使用最小欧几里得距离(默认值)计算样本质量

>>> import numpy as np
>>> from scipy.stats import qmc
>>> rng = np.random.default_rng()
>>> sample = qmc.LatinHypercube(d=2, seed=rng).random(50)
>>> qmc.geometric_discrepancy(sample)
0.03708161435687876

使用最小生成树中的平均边长计算质量

>>> qmc.geometric_discrepancy(sample, method='mst')
0.1105149978798376

显示最小生成树和距离最小的点

>>> import matplotlib.pyplot as plt
>>> from matplotlib.lines import Line2D
>>> from scipy.sparse.csgraph import minimum_spanning_tree
>>> from scipy.spatial.distance import pdist, squareform
>>> dist = pdist(sample)
>>> mst = minimum_spanning_tree(squareform(dist))
>>> edges = np.where(mst.toarray() > 0)
>>> edges = np.asarray(edges).T
>>> min_dist = np.min(dist)
>>> min_idx = np.argwhere(squareform(dist) == min_dist)[0]
>>> fig, ax = plt.subplots(figsize=(10, 5))
>>> _ = ax.set(aspect='equal', xlabel=r'$x_1$', ylabel=r'$x_2$',
...            xlim=[0, 1], ylim=[0, 1])
>>> for edge in edges:
...     ax.plot(sample[edge, 0], sample[edge, 1], c='k')
>>> ax.scatter(sample[:, 0], sample[:, 1])
>>> ax.add_patch(plt.Circle(sample[min_idx[0]], min_dist, color='red', fill=False))
>>> markers = [
...     Line2D([0], [0], marker='o', lw=0, label='Sample points'),
...     Line2D([0], [0], color='k', label='Minimum spanning tree'),
...     Line2D([0], [0], marker='o', lw=0, markerfacecolor='w', markeredgecolor='r',
...            label='Minimum point-to-point distance'),
... ]
>>> ax.legend(handles=markers, loc='center left', bbox_to_anchor=(1, 0.5));
>>> plt.show()
../../_images/scipy-stats-qmc-geometric_discrepancy-1.png