IR2Vec
Loading...
Searching...
No Matches
OpenKE.module.model.Model.Model Class Reference
Inheritance diagram for OpenKE.module.model.Model.Model:
Collaboration diagram for OpenKE.module.model.Model.Model:

Public Member Functions

 __init__ (self, ent_tot, rel_tot)
 
 forward (self)
 
 predict (self)
 
- Public Member Functions inherited from OpenKE.module.BaseModule.BaseModule
 load_checkpoint (self, path)
 
 save_checkpoint (self, path)
 
 load_parameters (self, path)
 
 save_parameters (self, path)
 
 get_parameters (self, mode="numpy", param_dict=None)
 
 set_parameters (self, parameters)
 

Public Attributes

 ent_tot = ent_tot
 
 rel_tot = rel_tot
 
- Public Attributes inherited from OpenKE.module.BaseModule.BaseModule
 zero_const = nn.Parameter(torch.Tensor([0]))
 
 pi_const = nn.Parameter(torch.Tensor([3.14159265358979323846]))
 

Constructor & Destructor Documentation

◆ __init__()

OpenKE.module.model.Model.Model.__init__ ( self,
ent_tot,
rel_tot )

The documentation for this class was generated from the following file: