API¶
Python API
- pybind11_ke.utils
- pybind11_ke.data
- 平移模型和语义匹配模型数据加载器
- TrainDataSampler
- TrainDataLoader
TrainDataLoaderTrainDataLoader.__init__()TrainDataLoader.__iter__()TrainDataLoader.__len__()TrainDataLoader.__weakref__TrainDataLoader.batch_sizeTrainDataLoader.bernTrainDataLoader.cross_sampling()TrainDataLoader.ent_fileTrainDataLoader.ent_tolTrainDataLoader.get_batch_size()TrainDataLoader.get_ent_tol()TrainDataLoader.get_rel_tol()TrainDataLoader.get_train_tot()TrainDataLoader.in_pathTrainDataLoader.nbatchesTrainDataLoader.neg_entTrainDataLoader.neg_relTrainDataLoader.read()TrainDataLoader.rel_fileTrainDataLoader.rel_tolTrainDataLoader.sampling()TrainDataLoader.sampling_head()TrainDataLoader.sampling_modeTrainDataLoader.sampling_tail()TrainDataLoader.threadsTrainDataLoader.train_fileTrainDataLoader.train_tot
- TestDataSampler
- TestDataLoader
TestDataLoaderTestDataLoader.__init__()TestDataLoader.__iter__()TestDataLoader.__len__()TestDataLoader.__weakref__TestDataLoader.ent_fileTestDataLoader.ent_tolTestDataLoader.get_ent_tol()TestDataLoader.get_rel_tol()TestDataLoader.get_test_tol()TestDataLoader.get_valid_tol()TestDataLoader.in_pathTestDataLoader.read()TestDataLoader.rel_fileTestDataLoader.rel_tolTestDataLoader.sampling()TestDataLoader.sampling_modeTestDataLoader.set_sampling_mode()TestDataLoader.test_fileTestDataLoader.test_tolTestDataLoader.train_fileTestDataLoader.type_constrainTestDataLoader.valid_fileTestDataLoader.valid_tol
- 图神经网络模型数据加载器
- 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
- 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
- GraphSampler
GraphSamplerGraphSampler.__init__()GraphSampler.__weakref__GraphSampler.add_reverse_relation()GraphSampler.add_train_reverse_triples()GraphSampler.batch_sizeGraphSampler.build_graph()GraphSampler.comp_deg_norm()GraphSampler.ent2idGraphSampler.ent_fileGraphSampler.ent_tolGraphSampler.get_hr2t_rt2h_from_train()GraphSampler.get_hr_train()GraphSampler.get_id()GraphSampler.get_train()GraphSampler.get_train_triples_id()GraphSampler.hr2t_trainGraphSampler.id2entGraphSampler.id2relGraphSampler.in_pathGraphSampler.neg_entGraphSampler.node_norm_to_edge_norm()GraphSampler.rel2idGraphSampler.rel_fileGraphSampler.rel_tolGraphSampler.rt2h_trainGraphSampler.sampling()GraphSampler.sampling_negative()GraphSampler.sampling_positive()GraphSampler.train_fileGraphSampler.train_tolGraphSampler.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
- GraphTestSampler
GraphTestSamplerGraphTestSampler.__init__()GraphTestSampler.__weakref__GraphTestSampler.add_valid_test_reverse_triples()GraphTestSampler.all_true_triplesGraphTestSampler.ent_tolGraphTestSampler.get_all_true_triples()GraphTestSampler.get_hr2t_rt2h_from_all()GraphTestSampler.get_test()GraphTestSampler.get_valid()GraphTestSampler.get_valid_test_triples_id()GraphTestSampler.hr2t_allGraphTestSampler.powerGraphTestSampler.rt2h_allGraphTestSampler.samplerGraphTestSampler.sampling()GraphTestSampler.test_fileGraphTestSampler.test_tolGraphTestSampler.test_triplesGraphTestSampler.triplesGraphTestSampler.valid_fileGraphTestSampler.valid_tolGraphTestSampler.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_valid()CompGCNTestSampler.get_valid_test_triples_id()CompGCNTestSampler.hr2t_allCompGCNTestSampler.powerCompGCNTestSampler.rt2h_allCompGCNTestSampler.samplerCompGCNTestSampler.sampling()CompGCNTestSampler.test_fileCompGCNTestSampler.test_tolCompGCNTestSampler.test_triplesCompGCNTestSampler.triplesCompGCNTestSampler.valid_fileCompGCNTestSampler.valid_tolCompGCNTestSampler.valid_triples
- GraphDataLoader
GraphDataLoaderGraphDataLoader.__init__()GraphDataLoader.__weakref__GraphDataLoader.batch_sizeGraphDataLoader.data_testGraphDataLoader.data_trainGraphDataLoader.data_valGraphDataLoader.ent_fileGraphDataLoader.in_pathGraphDataLoader.neg_entGraphDataLoader.num_workersGraphDataLoader.rel_fileGraphDataLoader.testGraphDataLoader.test_batch_sizeGraphDataLoader.test_dataloader()GraphDataLoader.test_fileGraphDataLoader.test_samplerGraphDataLoader.train_dataloader()GraphDataLoader.train_fileGraphDataLoader.train_samplerGraphDataLoader.val_dataloader()GraphDataLoader.valid_file
- KGReader
- 超参数优化默认搜索范围
- 平移模型和语义匹配模型数据加载器
- 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.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.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.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.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.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.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.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.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.zero_const
- ComplEx
ComplExComplEx.__call__()ComplEx.__init__()ComplEx.__new__()ComplEx.__repr__()ComplEx.__weakref__ComplEx._calc()ComplEx.dimComplEx.ent_im_embeddingsComplEx.ent_re_embeddingsComplEx.ent_tolComplEx.forward()ComplEx.get_parameters()ComplEx.load_checkpoint()ComplEx.load_parameters()ComplEx.pi_constComplEx.predict()ComplEx.regularization()ComplEx.rel_im_embeddingsComplEx.rel_re_embeddingsComplEx.rel_tolComplEx.save_checkpoint()ComplEx.save_parameters()ComplEx.set_parameters()ComplEx.zero_const
- Analogy
AnalogyAnalogy.__call__()Analogy.__init__()Analogy.__new__()Analogy.__repr__()Analogy.__weakref__Analogy._calc()Analogy.dimAnalogy.ent_embeddingsAnalogy.ent_im_embeddingsAnalogy.ent_re_embeddingsAnalogy.ent_tolAnalogy.forward()Analogy.get_parameters()Analogy.load_checkpoint()Analogy.load_parameters()Analogy.pi_constAnalogy.predict()Analogy.regularization()Analogy.rel_embeddingsAnalogy.rel_im_embeddingsAnalogy.rel_re_embeddingsAnalogy.rel_tolAnalogy.save_checkpoint()Analogy.save_parameters()Analogy.set_parameters()Analogy.zero_const
- SimplE
SimplESimplE.__call__()SimplE.__init__()SimplE.__new__()SimplE.__repr__()SimplE.__weakref__SimplE.dimSimplE.ent_h_embeddingsSimplE.ent_t_embeddingsSimplE.ent_tolSimplE.forward()SimplE.get_parameters()SimplE.load_checkpoint()SimplE.load_parameters()SimplE.pi_constSimplE.predict()SimplE.regularization()SimplE.rel_embeddingsSimplE.rel_inv_embeddingsSimplE.rel_tolSimplE.save_checkpoint()SimplE.save_parameters()SimplE.set_parameters()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.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.predict()MarginLoss.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
- Cross_Entropy_Loss
Cross_Entropy_LossCross_Entropy_Loss.__call__()Cross_Entropy_Loss.__init__()Cross_Entropy_Loss.__new__()Cross_Entropy_Loss.__repr__()Cross_Entropy_Loss.__weakref__Cross_Entropy_Loss.forward()Cross_Entropy_Loss.get_parameters()Cross_Entropy_Loss.load_checkpoint()Cross_Entropy_Loss.load_parameters()Cross_Entropy_Loss.lossCross_Entropy_Loss.modelCross_Entropy_Loss.pi_constCross_Entropy_Loss.save_checkpoint()Cross_Entropy_Loss.save_parameters()Cross_Entropy_Loss.set_parameters()Cross_Entropy_Loss.zero_const
- MarginLoss
- 超参数优化默认搜索范围
- pybind11_ke.module.strategy
- 策略基类
- 策略子类
- NegativeSampling
NegativeSamplingNegativeSampling.__call__()NegativeSampling.__init__()NegativeSampling.__new__()NegativeSampling.__repr__()NegativeSampling.__weakref__NegativeSampling._get_negative_score()NegativeSampling._get_positive_score()NegativeSampling.batch_sizeNegativeSampling.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.configure_optimizers()Trainer.data_loaderTrainer.deltaTrainer.deviceTrainer.early_stoppingTrainer.epochsTrainer.get_model()Trainer.gpu_idTrainer.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
- GraphTrainer
GraphTrainerGraphTrainer.__init__()GraphTrainer.__weakref__GraphTrainer.configure_optimizers()GraphTrainer.data_loaderGraphTrainer.deltaGraphTrainer.deviceGraphTrainer.early_stoppingGraphTrainer.epochsGraphTrainer.get_model()GraphTrainer.gpu_idGraphTrainer.log_intervalGraphTrainer.lrGraphTrainer.metricGraphTrainer.modelGraphTrainer.opt_methodGraphTrainer.optimizerGraphTrainer.patienceGraphTrainer.print_test()GraphTrainer.run()GraphTrainer.save_intervalGraphTrainer.save_pathGraphTrainer.schedulerGraphTrainer.testGraphTrainer.testerGraphTrainer.to_var()GraphTrainer.train_one_step()GraphTrainer.use_early_stoppingGraphTrainer.use_gpuGraphTrainer.use_wandbGraphTrainer.valid_interval
- Trainer
- 评估循环
- Tester
- GraphTester
GraphTesterGraphTester.__init__()GraphTester.__weakref__GraphTester.data_loaderGraphTester.deviceGraphTester.modelGraphTester.predictionGraphTester.run_link_prediction()GraphTester.sampling_modeGraphTester.set_sampling_mode()GraphTester.test_dataloaderGraphTester.test_one_step()GraphTester.to_var()GraphTester.use_gpuGraphTester.val_dataloader
- 链接预测函数
- 并行训练函数
- 超参数优化默认搜索范围
- 超参数优化训练循环函数
- 训练循环