0001 function A = slaffinitymat(X, X2, nnparams, varargin)
0002
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012
0013
0014
0015
0016
0017
0018
0019
0020
0021
0022
0023
0024
0025
0026
0027
0028
0029
0030
0031
0032
0033
0034
0035
0036
0037
0038
0039
0040
0041
0042
0043
0044
0045
0046
0047
0048
0049
0050
0051
0052
0053
0054
0055
0056
0057
0058
0059
0060
0061
0062
0063
0064
0065
0066
0067
0068
0069
0070 if nargin < 3
0071 raise_lackinput('slaffinitymat', 3);
0072 end
0073
0074 opts.sparse = true;
0075 opts.kernel = 'heat';
0076 opts.diffusion = 1;
0077 opts.tfunctor = [];
0078 opts.sym = true;
0079 opts.symmethod = [];
0080 opts.excludeself = false;
0081 opts = slparseprops(opts, varargin{:});
0082
0083 if isempty(X2)
0084 if opts.excludeself
0085 X2 = [];
0086 else
0087 X2 = X;
0088 end
0089 else
0090 opts.sym = false;
0091 end
0092
0093 if isempty(opts.tfunctor)
0094 switch opts.kernel
0095 case 'heat'
0096 tfunctor = {@internal_compaffinity, opts.diffusion};
0097 case 'simple'
0098 tfunctor = @(x) ones(size(x));
0099 otherwise
0100 error('sltoolbox:invalidarg', ...
0101 'Invalid kernel name: %s', opts.kernel);
0102 end
0103 else
0104 tfunctor = opts.tfunctor;
0105 end
0106
0107
0108
0109 A = slnngraph(X, X2, nnparams, ...
0110 'valtype', 'numeric', ...
0111 'sparse', opts.sparse, ...
0112 'tfunctor', tfunctor, ...
0113 'sym', opts.sym, ...
0114 'symmethod', opts.symmethod);
0115
0116
0117
0118 function affvals = internal_compaffinity(dists, diffusion)
0119
0120 sqs = dists .* dists;
0121 sqs = sqs(:);
0122
0123 avgsq = sum(sqs) / length(sqs);
0124 s = (diffusion^2) * avgsq * 2;
0125
0126 affvals = exp(-sqs / s);
0127
0128
0129
0130
0131
0132
0133
0134
0135
0136
0137
0138
0139