Skip to main content
. Author manuscript; available in PMC: 2022 Oct 20.
Published in final edited form as: Int Conf Knowl Syst Eng. 2020 Dec 16;2020:281–286. doi: 10.1109/kse50997.2020.9287797

TABLE II.

Model parameters

Parameter Baseline Siamese KG-LSTM
Exp.1 Exp.2,3,4
Framework Keras 2.2.4 with Tensorflow backend 2.1.0
Word Vector Size 200
Maximum Input Length 30
Embedding BioWordVec Trainable BioWordVec + Node2Vec Trainable
LSTM Hidden Units 50
LSTM Activation Tanh
Dense Layer 1 - 128 units with ReLU activation
Dense Layer 2 - 50 units with ReLU activation
Weights and Biases Xavier Initialization
Optimizer Adam
Learning Rate 0.001
Loss Function Mean Squared Error (MSE)
Batch Size 256
Epochs 9
Train/Validation Ratio 80/20