API¶
Python API
- pybind11_ke.utils
- pybind11_ke.data
- 数据预加载器
- KGReader
KGReaderKGReader.__init__()KGReader.__weakref__KGReader.ent2idKGReader.ent_fileKGReader.ent_tolKGReader.get_hr2t_rt2h_from_train()KGReader.get_hr_train()KGReader.get_id()KGReader.get_train()KGReader.get_train_triples_id()KGReader.hr2t_trainKGReader.id2entKGReader.id2relKGReader.in_pathKGReader.rel2idKGReader.rel_fileKGReader.rel_tolKGReader.rt2h_trainKGReader.train_fileKGReader.train_tolKGReader.train_triples
- KGReader
- 平移模型和语义匹配模型训练集数据采样器
- TradSampler
TradSamplerTradSampler.__init__()TradSampler.__weakref__TradSampler.batch_sizeTradSampler.corrupt_head()TradSampler.corrupt_tail()TradSampler.ent2idTradSampler.ent_fileTradSampler.ent_tolTradSampler.get_hr2t_rt2h_from_train()TradSampler.get_hr_train()TradSampler.get_id()TradSampler.get_train()TradSampler.get_train_triples_id()TradSampler.hr2t_trainTradSampler.id2entTradSampler.id2relTradSampler.in_pathTradSampler.neg_entTradSampler.rel2idTradSampler.rel_fileTradSampler.rel_tolTradSampler.rt2h_trainTradSampler.sampling()TradSampler.train_fileTradSampler.train_tolTradSampler.train_triples
- UniSampler
UniSamplerUniSampler.__init__()UniSampler.__weakref__UniSampler.batch_sizeUniSampler.corrupt_head()UniSampler.corrupt_tail()UniSampler.ent2idUniSampler.ent_fileUniSampler.ent_tolUniSampler.get_hr2t_rt2h_from_train()UniSampler.get_hr_train()UniSampler.get_id()UniSampler.get_train()UniSampler.get_train_triples_id()UniSampler.head_batch()UniSampler.hr2t_trainUniSampler.id2entUniSampler.id2relUniSampler.in_pathUniSampler.neg_entUniSampler.rel2idUniSampler.rel_fileUniSampler.rel_tolUniSampler.rt2h_trainUniSampler.sampling()UniSampler.tail_batch()UniSampler.train_fileUniSampler.train_tolUniSampler.train_triples
- BernSampler
BernSamplerBernSampler.__init__()BernSampler.__normal_batch()BernSampler.__weakref__BernSampler.batch_sizeBernSampler.corrupt_head()BernSampler.corrupt_tail()BernSampler.ent2idBernSampler.ent_fileBernSampler.ent_tolBernSampler.get_hr2t_rt2h_from_train()BernSampler.get_hr_train()BernSampler.get_id()BernSampler.get_tph_hpt()BernSampler.get_train()BernSampler.get_train_triples_id()BernSampler.hr2t_trainBernSampler.id2entBernSampler.id2relBernSampler.in_pathBernSampler.neg_entBernSampler.rel2idBernSampler.rel_fileBernSampler.rel_tolBernSampler.rt2h_trainBernSampler.sampling()BernSampler.train_fileBernSampler.train_tolBernSampler.train_triples
- TradSampler
- 图神经网络模型训练集数据采样器
- RevSampler
RevSamplerRevSampler.__init__()RevSampler.__weakref__RevSampler.add_reverse_relation()RevSampler.add_train_reverse_triples()RevSampler.ent2idRevSampler.ent_fileRevSampler.ent_tolRevSampler.get_hr2t_rt2h_from_train()RevSampler.get_hr_train()RevSampler.get_id()RevSampler.get_train()RevSampler.get_train_triples_id()RevSampler.hr2t_trainRevSampler.id2entRevSampler.id2relRevSampler.in_pathRevSampler.rel2idRevSampler.rel_fileRevSampler.rel_tolRevSampler.rt2h_trainRevSampler.train_fileRevSampler.train_tolRevSampler.train_triples
- RGCNSampler
RGCNSamplerRGCNSampler.__init__()RGCNSampler.__weakref__RGCNSampler.add_reverse_relation()RGCNSampler.add_train_reverse_triples()RGCNSampler.batch_sizeRGCNSampler.build_graph()RGCNSampler.comp_deg_norm()RGCNSampler.ent2idRGCNSampler.ent_fileRGCNSampler.ent_tolRGCNSampler.get_hr2t_rt2h_from_train()RGCNSampler.get_hr_train()RGCNSampler.get_id()RGCNSampler.get_train()RGCNSampler.get_train_triples_id()RGCNSampler.hr2t_trainRGCNSampler.id2entRGCNSampler.id2relRGCNSampler.in_pathRGCNSampler.neg_entRGCNSampler.node_norm_to_edge_norm()RGCNSampler.rel2idRGCNSampler.rel_fileRGCNSampler.rel_tolRGCNSampler.rt2h_trainRGCNSampler.sampling()RGCNSampler.sampling_negative()RGCNSampler.sampling_positive()RGCNSampler.train_fileRGCNSampler.train_tolRGCNSampler.train_triples
- CompGCNSampler
CompGCNSamplerCompGCNSampler.__init__()CompGCNSampler.__weakref__CompGCNSampler.add_reverse_relation()CompGCNSampler.add_train_reverse_triples()CompGCNSampler.batch_sizeCompGCNSampler.build_graph()CompGCNSampler.comp_deg_norm()CompGCNSampler.ent2idCompGCNSampler.ent_fileCompGCNSampler.ent_tolCompGCNSampler.get_hr2t_rt2h_from_train()CompGCNSampler.get_hr_train()CompGCNSampler.get_id()CompGCNSampler.get_train()CompGCNSampler.get_train_triples_id()CompGCNSampler.hr2t_trainCompGCNSampler.id2entCompGCNSampler.id2relCompGCNSampler.in_pathCompGCNSampler.neg_entCompGCNSampler.node_norm_to_edge_norm()CompGCNSampler.rel2idCompGCNSampler.rel_fileCompGCNSampler.rel_tolCompGCNSampler.rt2h_trainCompGCNSampler.sampling()CompGCNSampler.sampling_negative()CompGCNSampler.sampling_positive()CompGCNSampler.train_fileCompGCNSampler.train_tolCompGCNSampler.train_triples
- RevSampler
- 测试集数据采样器
- TestSampler
TestSamplerTestSampler.__init__()TestSampler.__weakref__TestSampler.all_true_triplesTestSampler.ent_tolTestSampler.get_all_true_triples()TestSampler.get_hr2t_rt2h_from_all()TestSampler.get_test()TestSampler.get_type_constrain_id()TestSampler.get_valid()TestSampler.get_valid_test_triples_id()TestSampler.hr2t_allTestSampler.rel_headsTestSampler.rel_tailsTestSampler.rt2h_allTestSampler.samplerTestSampler.sampling()TestSampler.test_fileTestSampler.test_tolTestSampler.test_triplesTestSampler.type_constrainTestSampler.valid_fileTestSampler.valid_tolTestSampler.valid_triples
- TradTestSampler
TradTestSamplerTradTestSampler.__init__()TradTestSampler.__weakref__TradTestSampler.all_true_triplesTradTestSampler.ent_tolTradTestSampler.get_all_true_triples()TradTestSampler.get_hr2t_rt2h_from_all()TradTestSampler.get_test()TradTestSampler.get_type_constrain_id()TradTestSampler.get_valid()TradTestSampler.get_valid_test_triples_id()TradTestSampler.hr2t_allTradTestSampler.rel_headsTradTestSampler.rel_tailsTradTestSampler.rt2h_allTradTestSampler.samplerTradTestSampler.sampling()TradTestSampler.test_fileTradTestSampler.test_tolTradTestSampler.test_triplesTradTestSampler.type_constrainTradTestSampler.valid_fileTradTestSampler.valid_tolTradTestSampler.valid_triples
- RGCNTestSampler
RGCNTestSamplerRGCNTestSampler.__init__()RGCNTestSampler.__weakref__RGCNTestSampler.add_valid_test_reverse_triples()RGCNTestSampler.all_true_triplesRGCNTestSampler.ent_tolRGCNTestSampler.get_all_true_triples()RGCNTestSampler.get_hr2t_rt2h_from_all()RGCNTestSampler.get_test()RGCNTestSampler.get_type_constrain_id()RGCNTestSampler.get_valid()RGCNTestSampler.get_valid_test_triples_id()RGCNTestSampler.hr2t_allRGCNTestSampler.powerRGCNTestSampler.rel_headsRGCNTestSampler.rel_tailsRGCNTestSampler.rt2h_allRGCNTestSampler.samplerRGCNTestSampler.sampling()RGCNTestSampler.test_fileRGCNTestSampler.test_tolRGCNTestSampler.test_triplesRGCNTestSampler.triplesRGCNTestSampler.type_constrainRGCNTestSampler.valid_fileRGCNTestSampler.valid_tolRGCNTestSampler.valid_triples
- CompGCNTestSampler
CompGCNTestSamplerCompGCNTestSampler.__init__()CompGCNTestSampler.__weakref__CompGCNTestSampler.add_valid_test_reverse_triples()CompGCNTestSampler.all_true_triplesCompGCNTestSampler.ent_tolCompGCNTestSampler.get_all_true_triples()CompGCNTestSampler.get_hr2t_rt2h_from_all()CompGCNTestSampler.get_test()CompGCNTestSampler.get_type_constrain_id()CompGCNTestSampler.get_valid()CompGCNTestSampler.get_valid_test_triples_id()CompGCNTestSampler.hr2t_allCompGCNTestSampler.powerCompGCNTestSampler.rel_headsCompGCNTestSampler.rel_tailsCompGCNTestSampler.rt2h_allCompGCNTestSampler.samplerCompGCNTestSampler.sampling()CompGCNTestSampler.test_fileCompGCNTestSampler.test_tolCompGCNTestSampler.test_triplesCompGCNTestSampler.triplesCompGCNTestSampler.type_constrainCompGCNTestSampler.valid_fileCompGCNTestSampler.valid_tolCompGCNTestSampler.valid_triples
- TestSampler
- 数据加载器
- KGEDataLoader
KGEDataLoaderKGEDataLoader.__init__()KGEDataLoader.__weakref__KGEDataLoader.batch_sizeKGEDataLoader.data_testKGEDataLoader.data_trainKGEDataLoader.data_valKGEDataLoader.ent_fileKGEDataLoader.get_ent_tol()KGEDataLoader.get_rel_tol()KGEDataLoader.in_pathKGEDataLoader.neg_entKGEDataLoader.num_workersKGEDataLoader.rel_fileKGEDataLoader.testKGEDataLoader.test_batch_sizeKGEDataLoader.test_dataloader()KGEDataLoader.test_fileKGEDataLoader.test_samplerKGEDataLoader.train_dataloader()KGEDataLoader.train_fileKGEDataLoader.train_samplerKGEDataLoader.type_constrainKGEDataLoader.val_dataloader()KGEDataLoader.valid_file
- KGEDataLoader
- 超参数优化默认搜索范围
- 数据预加载器
- pybind11_ke.module
- 基础模块
- BaseModule
BaseModuleBaseModule.__call__()BaseModule.__init__()BaseModule.__new__()BaseModule.__repr__()BaseModule.__weakref__BaseModule.get_parameters()BaseModule.load_checkpoint()BaseModule.load_parameters()BaseModule.pi_constBaseModule.save_checkpoint()BaseModule.save_parameters()BaseModule.set_parameters()BaseModule.zero_const
- BaseModule
- 基础模块
- pybind11_ke.module.model
- 基础模块
- CompGCNCov
CompGCNCovCompGCNCov.__call__()CompGCNCov.__init__()CompGCNCov.__new__()CompGCNCov.__repr__()CompGCNCov.__weakref__CompGCNCov.actCompGCNCov.biasCompGCNCov.bnCompGCNCov.comp()CompGCNCov.dropCompGCNCov.forward()CompGCNCov.get_param()CompGCNCov.in_channelsCompGCNCov.loop_relCompGCNCov.message_func()CompGCNCov.opnCompGCNCov.out_channelsCompGCNCov.reduce_func()CompGCNCov.relCompGCNCov.w_rel
- CompGCNCov
- 模型基类
- Model
ModelModel.__call__()Model.__init__()Model.__new__()Model.__repr__()Model.__weakref__Model.ent_tolModel.forward()Model.get_parameters()Model.load_checkpoint()Model.load_parameters()Model.pi_constModel.predict()Model.rel_tolModel.save_checkpoint()Model.save_parameters()Model.set_parameters()Model.tri2emb()Model.zero_const
- Model
- 平移模型
- TransE
TransETransE.__call__()TransE.__init__()TransE.__new__()TransE.__repr__()TransE.__weakref__TransE._calc()TransE.dimTransE.ent_embeddingsTransE.ent_tolTransE.forward()TransE.get_parameters()TransE.load_checkpoint()TransE.load_parameters()TransE.marginTransE.norm_flagTransE.p_normTransE.pi_constTransE.predict()TransE.regularization()TransE.rel_embeddingsTransE.rel_tolTransE.save_checkpoint()TransE.save_parameters()TransE.set_parameters()TransE.tri2emb()TransE.zero_const
- TransH
TransHTransH.__call__()TransH.__init__()TransH.__new__()TransH.__repr__()TransH.__weakref__TransH._calc()TransH._transfer()TransH.dimTransH.ent_embeddingsTransH.ent_tolTransH.forward()TransH.get_parameters()TransH.load_checkpoint()TransH.load_parameters()TransH.marginTransH.norm_flagTransH.norm_vectorTransH.p_normTransH.pi_constTransH.predict()TransH.regularization()TransH.rel_embeddingsTransH.rel_tolTransH.save_checkpoint()TransH.save_parameters()TransH.set_parameters()TransH.tri2emb()TransH.zero_const
- TransR
TransRTransR.__call__()TransR.__init__()TransR.__new__()TransR.__repr__()TransR.__weakref__TransR._calc()TransR._transfer()TransR.dim_eTransR.dim_rTransR.ent_embeddingsTransR.ent_tolTransR.forward()TransR.get_parameters()TransR.load_checkpoint()TransR.load_parameters()TransR.marginTransR.norm_flagTransR.p_normTransR.pi_constTransR.predict()TransR.rand_initTransR.regularization()TransR.rel_embeddingsTransR.rel_tolTransR.save_checkpoint()TransR.save_parameters()TransR.set_parameters()TransR.transfer_matrixTransR.tri2emb()TransR.zero_const
- TransD
TransDTransD.__call__()TransD.__init__()TransD.__new__()TransD.__repr__()TransD.__weakref__TransD._calc()TransD._resize()TransD._transfer()TransD.dim_eTransD.dim_rTransD.ent_embeddingsTransD.ent_tolTransD.ent_transferTransD.forward()TransD.get_parameters()TransD.load_checkpoint()TransD.load_parameters()TransD.marginTransD.norm_flagTransD.p_normTransD.pi_constTransD.predict()TransD.regularization()TransD.rel_embeddingsTransD.rel_tolTransD.rel_transferTransD.save_checkpoint()TransD.save_parameters()TransD.set_parameters()TransD.tri2emb()TransD.tri2transfer()TransD.zero_const
- RotatE
RotatERotatE.__call__()RotatE.__init__()RotatE.__new__()RotatE.__repr__()RotatE.__weakref__RotatE._calc()RotatE.dim_eRotatE.dim_rRotatE.ent_embeddingsRotatE.ent_tolRotatE.epsilonRotatE.forward()RotatE.get_parameters()RotatE.load_checkpoint()RotatE.load_parameters()RotatE.marginRotatE.pi_constRotatE.predict()RotatE.regularization()RotatE.rel_embeddingsRotatE.rel_tolRotatE.save_checkpoint()RotatE.save_parameters()RotatE.set_parameters()RotatE.tri2emb()RotatE.zero_const
- TransE
- 语义匹配模型
- RESCAL
RESCALRESCAL.__call__()RESCAL.__init__()RESCAL.__new__()RESCAL.__repr__()RESCAL.__weakref__RESCAL._calc()RESCAL.dimRESCAL.ent_embeddingsRESCAL.ent_tolRESCAL.forward()RESCAL.get_parameters()RESCAL.load_checkpoint()RESCAL.load_parameters()RESCAL.pi_constRESCAL.predict()RESCAL.regularization()RESCAL.rel_matricesRESCAL.rel_tolRESCAL.save_checkpoint()RESCAL.save_parameters()RESCAL.set_parameters()RESCAL.tri2emb()RESCAL.zero_const
- DistMult
DistMultDistMult.__call__()DistMult.__init__()DistMult.__new__()DistMult.__repr__()DistMult.__weakref__DistMult._calc()DistMult.dimDistMult.ent_embeddingsDistMult.ent_tolDistMult.forward()DistMult.get_parameters()DistMult.l3_regularization()DistMult.load_checkpoint()DistMult.load_parameters()DistMult.pi_constDistMult.predict()DistMult.regularization()DistMult.rel_embeddingsDistMult.rel_tolDistMult.save_checkpoint()DistMult.save_parameters()DistMult.set_parameters()DistMult.tri2emb()DistMult.zero_const
- HolE
HolEHolE.__call__()HolE.__init__()HolE.__new__()HolE.__repr__()HolE.__weakref__HolE._calc()HolE._ccorr()HolE.dimHolE.ent_embeddingsHolE.ent_tolHolE.forward()HolE.get_parameters()HolE.l3_regularization()HolE.load_checkpoint()HolE.load_parameters()HolE.pi_constHolE.predict()HolE.regularization()HolE.rel_embeddingsHolE.rel_tolHolE.save_checkpoint()HolE.save_parameters()HolE.set_parameters()HolE.tri2emb()HolE.zero_const
- ComplEx
ComplExComplEx.__call__()ComplEx.__init__()ComplEx.__new__()ComplEx.__repr__()ComplEx.__weakref__ComplEx._calc()ComplEx.dimComplEx.ent_embeddingsComplEx.ent_tolComplEx.forward()ComplEx.get_parameters()ComplEx.load_checkpoint()ComplEx.load_parameters()ComplEx.pi_constComplEx.predict()ComplEx.regularization()ComplEx.rel_embeddingsComplEx.rel_tolComplEx.save_checkpoint()ComplEx.save_parameters()ComplEx.set_parameters()ComplEx.tri2emb()ComplEx.zero_const
- Analogy
AnalogyAnalogy.__call__()Analogy.__init__()Analogy.__new__()Analogy.__repr__()Analogy.__weakref__Analogy._calc()Analogy.dimAnalogy.ent_embeddingsAnalogy.ent_tolAnalogy.forward()Analogy.get_parameters()Analogy.load_checkpoint()Analogy.load_parameters()Analogy.pi_constAnalogy.predict()Analogy.regularization()Analogy.rel_embeddingsAnalogy.rel_tolAnalogy.save_checkpoint()Analogy.save_parameters()Analogy.set_parameters()Analogy.tri2emb()Analogy.zero_const
- SimplE
SimplESimplE.__call__()SimplE.__init__()SimplE.__new__()SimplE.__repr__()SimplE.__weakref__SimplE._calc()SimplE.dimSimplE.ent_embeddingsSimplE.ent_tolSimplE.forward()SimplE.get_parameters()SimplE.load_checkpoint()SimplE.load_parameters()SimplE.pi_constSimplE.predict()SimplE.regularization()SimplE.rel_embeddingsSimplE.rel_tolSimplE.save_checkpoint()SimplE.save_parameters()SimplE.set_parameters()SimplE.tri2emb()SimplE.zero_const
- RESCAL
- 图神经网络模型
- RGCN
RGCNRGCN.Loss_embRGCN.RGCNRGCN.__call__()RGCN.__init__()RGCN.__new__()RGCN.__repr__()RGCN.__weakref__RGCN.build_hidden_layer()RGCN.build_model()RGCN.dimRGCN.distmult_score_func()RGCN.ent_embRGCN.ent_tolRGCN.forward()RGCN.get_parameters()RGCN.load_checkpoint()RGCN.load_parameters()RGCN.num_layersRGCN.pi_constRGCN.predict()RGCN.rel_embRGCN.rel_tolRGCN.save_checkpoint()RGCN.save_parameters()RGCN.set_parameters()RGCN.tri2emb()RGCN.zero_const
- CompGCN
CompGCNCompGCN.GraphCovCompGCN.__call__()CompGCN.__init__()CompGCN.__new__()CompGCN.__repr__()CompGCN.__weakref__CompGCN.biasCompGCN.bn0CompGCN.bn1CompGCN.bn2CompGCN.concat()CompGCN.conv1CompGCN.conve()CompGCN.decoder_modelCompGCN.dimCompGCN.distmult()CompGCN.dropCompGCN.ent_embCompGCN.ent_tolCompGCN.fcCompGCN.fet_dropCompGCN.forward()CompGCN.get_parameters()CompGCN.hid_dropCompGCN.load_checkpoint()CompGCN.load_parameters()CompGCN.marginCompGCN.opnCompGCN.pi_constCompGCN.predict()CompGCN.rel_embCompGCN.rel_tolCompGCN.save_checkpoint()CompGCN.save_parameters()CompGCN.set_parameters()CompGCN.transe()CompGCN.tri2emb()CompGCN.zero_const
- RGCN
- 平移模型超参数优化默认搜索范围
- 语义匹配模型超参数优化默认搜索范围
- 图神经网络模型超参数优化默认搜索范围
- 基础模块
- pybind11_ke.module.loss
- 损失函数基类
- 损失函数子类
- MarginLoss
MarginLossMarginLoss.__call__()MarginLoss.__init__()MarginLoss.__new__()MarginLoss.__repr__()MarginLoss.__weakref__MarginLoss.adv_flagMarginLoss.adv_temperatureMarginLoss.forward()MarginLoss.get_parameters()MarginLoss.get_weights()MarginLoss.load_checkpoint()MarginLoss.load_parameters()MarginLoss.marginMarginLoss.pi_constMarginLoss.save_checkpoint()MarginLoss.save_parameters()MarginLoss.set_parameters()MarginLoss.zero_const
- SigmoidLoss
SigmoidLossSigmoidLoss.__call__()SigmoidLoss.__init__()SigmoidLoss.__new__()SigmoidLoss.__repr__()SigmoidLoss.__weakref__SigmoidLoss.adv_flagSigmoidLoss.adv_temperatureSigmoidLoss.criterionSigmoidLoss.forward()SigmoidLoss.get_parameters()SigmoidLoss.get_weights()SigmoidLoss.load_checkpoint()SigmoidLoss.load_parameters()SigmoidLoss.pi_constSigmoidLoss.predict()SigmoidLoss.save_checkpoint()SigmoidLoss.save_parameters()SigmoidLoss.set_parameters()SigmoidLoss.zero_const
- SoftplusLoss
SoftplusLossSoftplusLoss.__call__()SoftplusLoss.__init__()SoftplusLoss.__new__()SoftplusLoss.__repr__()SoftplusLoss.__weakref__SoftplusLoss.adv_flagSoftplusLoss.adv_temperatureSoftplusLoss.criterionSoftplusLoss.forward()SoftplusLoss.get_parameters()SoftplusLoss.get_weights()SoftplusLoss.load_checkpoint()SoftplusLoss.load_parameters()SoftplusLoss.pi_constSoftplusLoss.predict()SoftplusLoss.save_checkpoint()SoftplusLoss.save_parameters()SoftplusLoss.set_parameters()SoftplusLoss.zero_const
- RGCNLoss
RGCNLossRGCNLoss.__call__()RGCNLoss.__init__()RGCNLoss.__new__()RGCNLoss.__repr__()RGCNLoss.__weakref__RGCNLoss.forward()RGCNLoss.get_parameters()RGCNLoss.load_checkpoint()RGCNLoss.load_parameters()RGCNLoss.modelRGCNLoss.pi_constRGCNLoss.reg_loss()RGCNLoss.regularizationRGCNLoss.save_checkpoint()RGCNLoss.save_parameters()RGCNLoss.set_parameters()RGCNLoss.zero_const
- CompGCNLoss
CompGCNLossCompGCNLoss.__call__()CompGCNLoss.__init__()CompGCNLoss.__new__()CompGCNLoss.__repr__()CompGCNLoss.__weakref__CompGCNLoss.forward()CompGCNLoss.get_parameters()CompGCNLoss.load_checkpoint()CompGCNLoss.load_parameters()CompGCNLoss.lossCompGCNLoss.modelCompGCNLoss.pi_constCompGCNLoss.save_checkpoint()CompGCNLoss.save_parameters()CompGCNLoss.set_parameters()CompGCNLoss.zero_const
- MarginLoss
- 超参数优化默认搜索范围
- pybind11_ke.module.strategy
- 策略基类
- 策略子类
- NegativeSampling
NegativeSamplingNegativeSampling.__call__()NegativeSampling.__init__()NegativeSampling.__new__()NegativeSampling.__repr__()NegativeSampling.__weakref__NegativeSampling.forward()NegativeSampling.get_parameters()NegativeSampling.l3_regul_rateNegativeSampling.load_checkpoint()NegativeSampling.load_parameters()NegativeSampling.lossNegativeSampling.modelNegativeSampling.pi_constNegativeSampling.regul_rateNegativeSampling.save_checkpoint()NegativeSampling.save_parameters()NegativeSampling.set_parameters()NegativeSampling.zero_const
- RGCNSampling
RGCNSamplingRGCNSampling.__call__()RGCNSampling.__init__()RGCNSampling.__new__()RGCNSampling.__repr__()RGCNSampling.__weakref__RGCNSampling.forward()RGCNSampling.get_parameters()RGCNSampling.load_checkpoint()RGCNSampling.load_parameters()RGCNSampling.lossRGCNSampling.modelRGCNSampling.pi_constRGCNSampling.save_checkpoint()RGCNSampling.save_parameters()RGCNSampling.set_parameters()RGCNSampling.zero_const
- CompGCNSampling
CompGCNSamplingCompGCNSampling.__call__()CompGCNSampling.__init__()CompGCNSampling.__new__()CompGCNSampling.__repr__()CompGCNSampling.__weakref__CompGCNSampling.ent_tolCompGCNSampling.forward()CompGCNSampling.get_parameters()CompGCNSampling.load_checkpoint()CompGCNSampling.load_parameters()CompGCNSampling.lossCompGCNSampling.modelCompGCNSampling.pi_constCompGCNSampling.save_checkpoint()CompGCNSampling.save_parameters()CompGCNSampling.set_parameters()CompGCNSampling.smoothingCompGCNSampling.zero_const
- NegativeSampling
- 超参数优化默认搜索范围
- pybind11_ke.config
- 训练循环
- Trainer
TrainerTrainer.__init__()Trainer.__weakref__Trainer.acceleratorTrainer.configure_optimizers()Trainer.data_loaderTrainer.deltaTrainer.deviceTrainer.early_stoppingTrainer.epochsTrainer.get_device()Trainer.get_model()Trainer.is_local_main_process()Trainer.log_intervalTrainer.lrTrainer.metricTrainer.modelTrainer.opt_methodTrainer.optimizerTrainer.patienceTrainer.print_test()Trainer.run()Trainer.save_intervalTrainer.save_pathTrainer.schedulerTrainer.testTrainer.testerTrainer.to_var()Trainer.train_one_step()Trainer.use_early_stoppingTrainer.use_gpuTrainer.use_wandbTrainer.valid_interval
- Trainer
- 评估循环
- 链接预测函数
- 并行训练函数
- 超参数优化默认搜索范围
- 超参数优化训练循环函数
- 训练循环