unKR
- Data
- base_data_module
- DataPreprocess
GMUCBaseSamplerGMUCDataGMUCData.argsGMUCData.ent2idGMUCData.rel2idGMUCData.symbol2idGMUCData.e1rel_e2GMUCData.rele1_e2GMUCData.train_tasksGMUCData.dev_tasksGMUCData.test_tasksGMUCData.task_poolGMUCData.path_graphGMUCData.type2entsGMUCData.known_relsGMUCData.num_tasksGMUCData.rel2candidatesGMUCData.ent2icGMUCData.rel_uc1GMUCData.rel_uc2GMUCData.connectionsGMUCData.e1_rele2GMUCData.e1_degreesGMUCData.build_graph()GMUCData.get_e1rel_e2()GMUCData.get_ontology()GMUCData.get_rel2candidates()GMUCData.get_rele1_e2()GMUCData.get_tasks()
UKGDataUKGData.argsUKGData.ent2idUKGData.rel2idUKGData.id2entUKGData.id2relUKGData.train_triplesUKGData.valid_triplesUKGData.test_triplesUKGData.PSL_triplesUKGData.pseudo_triplesUKGData.all_true_triplesUKGData.hr2t_trainUKGData.rt2h_trainUKGData.h2rt_trainUKGData.t2rh_trainUKGData.hr2t_totalUKGData.rt2h_totalUKGData.RatioOfPSLUKGData.pseudo_dataiterUKGData.count_frequency()UKGData.get_h2rt_t2hr_from_train()UKGData.get_hr2t_rt2h_from_total()UKGData.get_hr2t_rt2h_from_train()UKGData.get_hr_train()UKGData.get_id()
UKGEBaseSampler
- Sampler
GMUCSamplerGMUCTestSamplerKGDatasetUKGEPSLSamplerUKGETestSamplerUKGETestSampler.samplerUKGETestSampler.hr2t_allUKGETestSampler.rt2h_allUKGETestSampler.hr2t_all_high_scoreUKGETestSampler.rt2h_all_high_scoreUKGETestSampler.num_entUKGETestSampler.construct_hr_map()UKGETestSampler.get_hr2t_rt2h_from_all()UKGETestSampler.get_sampling_keys()UKGETestSampler.sampling()
UKGEUniSamplernormal()
- KGDataModule
- Eval_Task
- Lit_Model
- BaseLitModel
BaseLitModelBaseLitModel.add_to_argparse()BaseLitModel.configure_optimizers()BaseLitModel.convert_neg_samples_batch()BaseLitModel.convert_semi_samples_batch()BaseLitModel.corrupt_head()BaseLitModel.corrupt_tail()BaseLitModel.forward()BaseLitModel.get_results()BaseLitModel.head_batch()BaseLitModel.select_random_elements()BaseLitModel.tail_batch()BaseLitModel.test_step()BaseLitModel.trainingBaseLitModel.training_step()BaseLitModel.validation_step()
Config
- BEUrRELitModel
BEUrRELitModelBEUrRELitModel.add_to_argparse()BEUrRELitModel.allow_zero_length_dataloader_with_multiple_devicesBEUrRELitModel.configure_optimizers()BEUrRELitModel.forward()BEUrRELitModel.precisionBEUrRELitModel.prepare_data_per_nodeBEUrRELitModel.test_epoch_end()BEUrRELitModel.test_step()BEUrRELitModel.trainingBEUrRELitModel.training_step()BEUrRELitModel.use_ampBEUrRELitModel.validation_epoch_end()BEUrRELitModel.validation_step()
- FocusELitModel
- GMUCLitModel
GMUCLitModelGMUCLitModel.add_to_argparse()GMUCLitModel.allow_zero_length_dataloader_with_multiple_devicesGMUCLitModel.configure_optimizers()GMUCLitModel.forward()GMUCLitModel.get_results()GMUCLitModel.precisionGMUCLitModel.prepare_data_per_nodeGMUCLitModel.test_epoch_end()GMUCLitModel.test_step()GMUCLitModel.trainingGMUCLitModel.training_step()GMUCLitModel.use_ampGMUCLitModel.val_test()GMUCLitModel.validation_epoch_end()GMUCLitModel.validation_step()
- GTransELitModel
GTransELitModelGTransELitModel.add_to_argparse()GTransELitModel.allow_zero_length_dataloader_with_multiple_devicesGTransELitModel.configure_optimizers()GTransELitModel.forward()GTransELitModel.precisionGTransELitModel.prepare_data_per_nodeGTransELitModel.test_epoch_end()GTransELitModel.test_step()GTransELitModel.trainingGTransELitModel.training_step()GTransELitModel.use_ampGTransELitModel.validation_epoch_end()GTransELitModel.validation_step()
- PASSLEAFLitModel
PASSLEAFLitModelPASSLEAFLitModel.add_to_argparse()PASSLEAFLitModel.allow_zero_length_dataloader_with_multiple_devicesPASSLEAFLitModel.configure_optimizers()PASSLEAFLitModel.forward()PASSLEAFLitModel.precisionPASSLEAFLitModel.prepare_data_per_nodePASSLEAFLitModel.test_epoch_end()PASSLEAFLitModel.test_step()PASSLEAFLitModel.trainingPASSLEAFLitModel.training_epoch_end()PASSLEAFLitModel.training_step()PASSLEAFLitModel.use_ampPASSLEAFLitModel.validation_epoch_end()PASSLEAFLitModel.validation_step()
- UKGELitModel
UKGELitModelUKGELitModel.add_to_argparse()UKGELitModel.allow_zero_length_dataloader_with_multiple_devicesUKGELitModel.configure_optimizers()UKGELitModel.forward()UKGELitModel.precisionUKGELitModel.prepare_data_per_nodeUKGELitModel.test_epoch_end()UKGELitModel.test_step()UKGELitModel.trainingUKGELitModel.training_step()UKGELitModel.use_ampUKGELitModel.validation_epoch_end()UKGELitModel.validation_step()
- UKGEPSLLitModel
UKGEPSLLitModelUKGEPSLLitModel.add_to_argparse()UKGEPSLLitModel.allow_zero_length_dataloader_with_multiple_devicesUKGEPSLLitModel.configure_optimizers()UKGEPSLLitModel.forward()UKGEPSLLitModel.precisionUKGEPSLLitModel.prepare_data_per_nodeUKGEPSLLitModel.test_epoch_end()UKGEPSLLitModel.test_step()UKGEPSLLitModel.trainingUKGEPSLLitModel.training_step()UKGEPSLLitModel.use_ampUKGEPSLLitModel.validation_epoch_end()UKGEPSLLitModel.validation_step()
- UKGsELitModel
UKGsELitModelUKGsELitModel.add_to_argparse()UKGsELitModel.allow_zero_length_dataloader_with_multiple_devicesUKGsELitModel.configure_optimizers()UKGsELitModel.forward()UKGsELitModel.precisionUKGsELitModel.prepare_data_per_nodeUKGsELitModel.test_epoch_end()UKGsELitModel.test_step()UKGsELitModel.trainingUKGsELitModel.training_step()UKGsELitModel.use_ampUKGsELitModel.validation_epoch_end()UKGsELitModel.validation_step()
- UPGATLitModel
UPGATLitModelUPGATLitModel.add_to_argparse()UPGATLitModel.allow_zero_length_dataloader_with_multiple_devicesUPGATLitModel.configure_optimizers()UPGATLitModel.forward()UPGATLitModel.precisionUPGATLitModel.prepare_data_per_nodeUPGATLitModel.test_epoch_end()UPGATLitModel.test_step()UPGATLitModel.trainingUPGATLitModel.training_step()UPGATLitModel.use_ampUPGATLitModel.validation_epoch_end()UPGATLitModel.validation_step()
- BaseLitModel
- loss
- Model