Input own similarity matrix?

[ameya98]

Thanks for the implementation. We have a similarity matrix W, and we were wondering if we could use SpectralNet to learn embeddings (in the spectral basis of the associated Laplacian of the similarity matrix).
It seems that W is being learned as part of the optimization procedure in SpectralNet, but we want to shortcircuit that.

[lihenryhfl]

Hi, you can actually create your own n x n affinity matrix as a function of the n inputs to the affinity matrix per batch. See

SpectralNet/src/core/networks.py

Line 108 in 43b0fca

# create affinity matrix

The easiest reference function you can look at is full_affinity(X, scale), which is used in the first part of the conditional in the referenced part of the python script.