IR2Vec
Loading...
Searching...
No Matches
OpenKE.config.Tester.Tester Class Reference
Inheritance diagram for OpenKE.config.Tester.Tester:
Collaboration diagram for OpenKE.config.Tester.Tester:

Public Member Functions

 __init__ (self, model=None, data_loader=None, use_gpu=True)
 
 set_model (self, model)
 
 set_data_loader (self, data_loader)
 
 set_use_gpu (self, use_gpu)
 
 to_var (self, x, use_gpu)
 
 test_one_step (self, data)
 
 random_sample (self, data_loader, sample_size)
 
 run_link_prediction (self, type_constrain=False, sample_size=None, sample_per=None)
 
 get_best_threshlod (self, score, ans)
 
 run_triple_classification (self, threshlod=None)
 

Public Attributes

 lib = ctypes.cdll.LoadLibrary(base_file)
 
 model = model
 
 data_loader = data_loader
 
 use_gpu = use_gpu
 

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