embedding.py 文件源码

python
阅读 29 收藏 0 点赞 0 评论 0

项目:sdp_kmeans 作者: simonsfoundation 项目源码 文件源码
def spectral_embedding(mat, target_dim, gramian=True, discard_first=True):
    if discard_first:
        last = -1
        first = target_dim - last
    else:
        first = target_dim
        last = None
    if not gramian:
        mat = mat.dot(mat.T)
    eigvals, eigvecs = eigh(mat)

    sl = slice(-first, last)
    eigvecs = eigvecs[:, sl]
    eigvals_crop = eigvals[sl]
    Y = eigvecs.dot(np.diag(np.sqrt(eigvals_crop)))
    Y = Y[:, ::-1]

    variance_explaned(eigvals, eigvals_crop)
    return Y
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号