__init__(self, ent_tot, rel_tot, dim=100) (defined in OpenKE.module.model.ComplEx.ComplEx) | OpenKE.module.model.ComplEx.ComplEx | |
_calc(self, h_re, h_im, t_re, t_im, r_re, r_im) (defined in OpenKE.module.model.ComplEx.ComplEx) | OpenKE.module.model.ComplEx.ComplEx | protected |
dim (defined in OpenKE.module.model.ComplEx.ComplEx) | OpenKE.module.model.ComplEx.ComplEx | |
ent_im_embeddings (defined in OpenKE.module.model.ComplEx.ComplEx) | OpenKE.module.model.ComplEx.ComplEx | |
ent_re_embeddings (defined in OpenKE.module.model.ComplEx.ComplEx) | OpenKE.module.model.ComplEx.ComplEx | |
ent_tot (defined in OpenKE.module.model.Model.Model) | OpenKE.module.model.Model.Model | |
forward(self, data) (defined in OpenKE.module.model.ComplEx.ComplEx) | OpenKE.module.model.ComplEx.ComplEx | |
get_parameters(self, mode="numpy", param_dict=None) (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |
load_checkpoint(self, path) (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |
load_parameters(self, path) (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |
pi_const (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |
predict(self, data) (defined in OpenKE.module.model.ComplEx.ComplEx) | OpenKE.module.model.ComplEx.ComplEx | |
regularization(self, data) (defined in OpenKE.module.model.ComplEx.ComplEx) | OpenKE.module.model.ComplEx.ComplEx | |
rel_im_embeddings (defined in OpenKE.module.model.ComplEx.ComplEx) | OpenKE.module.model.ComplEx.ComplEx | |
rel_re_embeddings (defined in OpenKE.module.model.ComplEx.ComplEx) | OpenKE.module.model.ComplEx.ComplEx | |
rel_tot (defined in OpenKE.module.model.Model.Model) | OpenKE.module.model.Model.Model | |
save_checkpoint(self, path) (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |
save_parameters(self, path) (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |
set_parameters(self, parameters) (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |
zero_const (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |