Table 3 ComplEx, DistMult, HolE, TransE, TransR hyperparameters
From: Gene expression knowledge graph for patient representation and diabetes prediction
Hyperparameter | ComplEx | DistMult | HolE | TransE | TransR |
|---|---|---|---|---|---|
Embedding size | 100 | 100 | 100 | 100 | 100 |
Optimization | Adagrad | Adagrad | Adagrad | SGD | SGD |
Train times | 500 | 500 | 500 | 500 | 500 |
Number batches | 100 | 100 | 100 | 100 | 100 |
Entity neg rate | 1 | 1 | 1 | 1 | 1 |
Relation neg rate | 0 | 0 | 0 | 0 | 0 |
Bern | 1 | 1 | 0 | 0 | 0 |
Alpha | 0.5 | 0.5 | 0.1 | 0.001 | 0.001 |
Lambda | 0.05 | 0.05 | – | – | 4 |
Margin | – | – | – | 1 | 1 |