__init__(self, ent_tot, rel_tot, dim=100, margin=6.0, epsilon=2.0) (defined in OpenKE.module.model.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | |
_calc(self, h, t, r, mode) (defined in OpenKE.module.model.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | protected |
dim_e (defined in OpenKE.module.model.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | |
dim_r (defined in OpenKE.module.model.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | |
ent_embedding_range (defined in OpenKE.module.model.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | |
ent_embeddings (defined in OpenKE.module.model.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | |
ent_tot (defined in OpenKE.module.model.Model.Model) | OpenKE.module.model.Model.Model | |
epsilon (defined in OpenKE.module.model.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | |
forward(self, data) (defined in OpenKE.module.model.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | |
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.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | |
pi_const (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |
predict(self, data) (defined in OpenKE.module.model.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | |
regularization(self, data) (defined in OpenKE.module.model.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | |
rel_embedding_range (defined in OpenKE.module.model.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | |
rel_embeddings (defined in OpenKE.module.model.RotatE.RotatE) | OpenKE.module.model.RotatE.RotatE | |
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 | |