Pytorch--模型搭建思路

参考代码
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,?>
----开始训练----