__init__(self, ent_tot, rel_tot, dim_e=100, dim_r=100, p_norm=1, norm_flag=True, rand_init=False, margin=None) (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
_calc(self, h, t, r, mode) (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | protected |
_transfer(self, e, r_transfer) (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | protected |
dim_e (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
dim_r (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
ent_embeddings (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
ent_tot (defined in OpenKE.module.model.Model.Model) | OpenKE.module.model.Model.Model | |
forward(self, data) (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
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.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
margin_flag (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
norm_flag (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
p_norm (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
pi_const (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |
predict(self, data) (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
rand_init (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
regularization(self, data) (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
rel_embeddings (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
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 | |
transfer_matrix (defined in OpenKE.module.model.TransR.TransR) | OpenKE.module.model.TransR.TransR | |
zero_const (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |