__init__(self, ent_tot, rel_tot, dim_e=100, dim_r=100, p_norm=1, norm_flag=True, margin=None, epsilon=None) (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
_calc(self, h, t, r, mode) (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | protected |
_resize(self, tensor, axis, size) (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | protected |
_transfer(self, e, e_transfer, r_transfer) (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | protected |
dim_e (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
dim_r (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
ent_embedding_range (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
ent_embeddings (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
ent_tot (defined in OpenKE.module.model.Model.Model) | OpenKE.module.model.Model.Model | |
ent_transfer (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
epsilon (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
forward(self, data) (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
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 | |
margin (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
margin_flag (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
norm_flag (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
p_norm (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
pi_const (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |
predict(self, data) (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
regularization(self, data) (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
rel_embedding_range (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
rel_embeddings (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
rel_tot (defined in OpenKE.module.model.Model.Model) | OpenKE.module.model.Model.Model | |
rel_transfer (defined in OpenKE.module.model.TransD.TransD) | OpenKE.module.model.TransD.TransD | |
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 | |