Ансамбль деревьев
作者
function predictor_association(trees, X)
n_features = size(X, 2)
n_trees = length(trees)
association = zeros(n_features, n_features)
for i in 1:n_features
association[i, i] = 1.0
end
for i in 1:n_features
for j in i+1:n_features
valid_idx = .!any(isnan, X[:, [i, j]]; dims=2)[:]
if sum(valid_idx) > 2
cor_val = abs(cor(X[valid_idx, i], X[valid_idx, j]))
association[i, j] = cor_val
association[j, i] = cor_val
end
end
end
return association
end