Stress属性 - sklearn.manifold.MDS/Python

pce*_*con 6 python machine-learning stress-testing scikit-learn mds

我正在使用scikit-learn方法MDS在某些数据中执行降维.我想检查压力值以获得减少的质量.我期待0到1之间的东西.但是,我得到的值超出了这个范围.这是一个最小的例子:

%matplotlib inline

from sklearn.preprocessing import normalize
from sklearn import manifold
from matplotlib import pyplot as plt
from matplotlib.lines import Line2D

import numpy


def similarity_measure(vec1, vec2):
    vec1_x = numpy.arctan2(vec1[1], vec1[0])
    vec2_x = numpy.arctan2(vec2[1], vec2[0])
    vec1_y = numpy.sqrt(numpy.sum(vec1[0] * vec1[0] + vec1[1] * vec1[1]))
    vec2_y = numpy.sqrt(numpy.sum(vec2[0] * vec2[0] + vec2[1] * vec2[1]))

    dot  = numpy.sum(vec1_x * vec2_x + vec1_y * vec2_y)
    mag1 = numpy.sqrt(numpy.sum(vec1_x * vec1_x + vec1_y * vec1_y))
    mag2 = numpy.sqrt(numpy.sum(vec2_x * vec2_x + vec2_y * vec2_y))
    return dot / (mag1 * mag2)

plt.figure(figsize=(15, 15))

delta = numpy.zeros((100, 100))
data_x = numpy.random.randint(0, 100, (100, 100))
data_y = numpy.random.randint(0, 100, (100, 100))

for j in range(100):
    for k in range(100):
        if j <= k:
            dist = similarity_measure((data_x[j].flatten(), data_y[j].flatten()), (data_x[k].flatten(), data_y[k].flatten()))
            delta[j, k] = delta[k, j] = dist

delta = 1-((delta+1)/2)  
delta /= numpy.max(delta)

mds = manifold.MDS(n_components=2, max_iter=3000, eps=1e-9, random_state=0,
               dissimilarity="precomputed", n_jobs=1)
coords = mds.fit(delta).embedding_
print mds.stress_

plt.scatter(coords[:, 0], coords[:, 1], marker='x', s=50, edgecolor='None')
plt.tight_layout()
Run Code Online (Sandbox Code Playgroud)

在我的测试中,打印了以下内容:

263.412196461

并产生了这个图像:

在此输入图像描述

如何在不知道最大值的情况下分析此值?或者如何将其标准化,使其在0和1之间?

谢谢.

小智 5

这是因为当前 scikit-learn 的实现计算并返回原始 Stress 值 (? r ),而您期望 Stress-1 (? 1 )。

前者的信息量并不大(其高值并不一定表示不合适),传达可靠性的更好方法是计算规范压力,例如。根据 Kruskal (1964, p. 3) 的 Stress-1 或多或少有以下解释:值 0 表示完美契合,0.025 优秀,0.05 良好,0.1 一般和 0.2 差。

我刚刚实现了 Stress-1 的计算并发送了 PR。同时,您可以使用此分支中的版本,其中当normalize参数设置为True(默认为False)时,使用并返回 Stress-1 而不是原始 Stress 。

欲了解更多信息,请参阅。Kruskal (1964, p. 8-9) 或 Borg and Groenen (2005, p. 41-43)。