参考代码
1.超参设置 main.py
--dataset FB15k-237 --num_iterations 500 --batch_size 128
--lr 0.0005 --dr 1.0 --edim 200 --rdim 200 --input_dropout 0.3
--hidden_dropout1 0.4 --hidden_dropout2 0.5 --label_smoothing 0.1
2.数据处理 load_data.py
self.train_data self.valid_data self.test_data
self.train_relations self.valid_relations self.test_relations
self.data
self.entities
self.relations
3.模型初始化
4.模型的训练
self.entity_idxs
self.relation_idxs
train_data_idxs # 训练数据集，id三元组
TuckER模型初始化
er_vocab # 训练数据集的数据对<h,r,t>
er_vocab_pairs # 训练数据集的数据对<h,r,?>
----开始训练----
