Skip to main content
. 2023 Mar 23;14:1159076. doi: 10.3389/fmicb.2023.1159076
Algorithm 1: Algorithm of our proposed method
Inputs: Known associations matrix Anm×nd, microbe similarity matrix KsmNm×Nm, disease similarity matrix Ksdnd×nd;
Output: The completed training of the generative network model
Step 1: Constructing the heterogeneous network Y(nm+nd)×(nd+nm) according to Formula (16);
Step 2: Input the feature matrix into the generative network, initializing Optimizer Parameter Information;
Step 3: for i=1Ndo (N is the number of training rounds of the generative adversarial network)
for l=1Ldo (L is the depth of the graph convolution model)
Compute the feature embedding of the L layer and output the generated prediction results
end for
Input the generated results and sample data into the decision network
Update optimizer parameter information
end for
Step 4: Save the model of the generative network