Research Article
Enhancing Text Generation via Parse Tree Embedding
Algorithm 1
Parse tree embedding VAE.
(1) | Initialize , , , with random weights , , , | (2) | M = number of Pre-train Parse tree encoder | (3) | N = number of training iterations | (4) | for each iteration do | (5) | Sample a sequence from dataset | (6) | Generate a Parse tree for based on by (1) | (7) | Given , generate embedding vector based on (2) | (8) | Compute the loss based on (3) | (9) | Compute the gradient of and | (10) | Update and parameters | (11) | end for | (12) | for each iteration do | (13) | Sample a sequence from dataset | (14) | Generate a parse tree for based on by (1) | (15) | Given generate embedding vector based on (2) | (16) | Given and , generate vector using | (17) | Compute the loss based on (4) | (18) | Computed the gradient of and | (19) | Update and parameters | (20) | end for |
|