Parameter | Unsupervised | Supervised |
---|---|---|
Learning rate | {2e−8, 2e−7, 2e−6, 2e−5} | {1e−4, 1e−3, 1e−2} |
Dimension | {64, 128, 256} | {64, 128, 256} |
Epochs | 1000 | 500 |
Sample Nodes | Layer1(9), Layer2(3) | Layer1(9), Layer2(3) |
Negative Sampling | 12 | NA |
Dropout | 0.1 | 0.1 |
Batch size | Task1 (1024), Task2 (2048) | Task1 (1024), Task2 (2048) |