__init__(self, adv_temperature=None, margin=6.0) (defined in OpenKE.module.loss.MarginLoss.MarginLoss) | OpenKE.module.loss.MarginLoss.MarginLoss | |
adv_flag (defined in OpenKE.module.loss.MarginLoss.MarginLoss) | OpenKE.module.loss.MarginLoss.MarginLoss | |
adv_temperature (defined in OpenKE.module.loss.MarginLoss.MarginLoss) | OpenKE.module.loss.MarginLoss.MarginLoss | |
forward(self, p_score, n_score) (defined in OpenKE.module.loss.MarginLoss.MarginLoss) | OpenKE.module.loss.MarginLoss.MarginLoss | |
get_parameters(self, mode="numpy", param_dict=None) (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |
get_weights(self, n_score) (defined in OpenKE.module.loss.MarginLoss.MarginLoss) | OpenKE.module.loss.MarginLoss.MarginLoss | |
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.loss.MarginLoss.MarginLoss) | OpenKE.module.loss.MarginLoss.MarginLoss | |
pi_const (defined in OpenKE.module.BaseModule.BaseModule) | OpenKE.module.BaseModule.BaseModule | |
predict(self, p_score, n_score) (defined in OpenKE.module.loss.MarginLoss.MarginLoss) | OpenKE.module.loss.MarginLoss.MarginLoss | |
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 | |