MCPcopy Create free account

hub / github.com/cheungdaven/QuatE / functions

Functions124 in github.com/cheungdaven/QuatE

↓ 8 callersMethodtest_one_step
(self, model, test_h, test_t, test_r)
config/Config.py:386
↓ 7 callersFunction_find
base/Corrupt.h:105
↓ 7 callersFunctionto_var
(x)
config/Config.py:20
↓ 5 callersMethodinit
(self)
config/Config.py:128
↓ 5 callersMethodset_alpha
(self, alpha)
config/Config.py:229
↓ 5 callersMethodset_bern
(self, bern)
config/Config.py:247
↓ 5 callersMethodset_checkpoint_dir
(self, checkpoint_dir)
config/Config.py:223
↓ 5 callersMethodset_dimension
(self, dim)
config/Config.py:250
↓ 5 callersMethodset_early_stopping_patience
(self, early_stopping_patience)
config/Config.py:279
↓ 5 callersMethodset_ent_neg_rate
(self, rate)
config/Config.py:267
↓ 5 callersMethodset_in_path
(self, in_path)
config/Config.py:208
↓ 5 callersMethodset_lmbda
(self, lmbda)
config/Config.py:232
↓ 5 callersMethodset_margin
(self, margin)
config/Config.py:205
↓ 5 callersMethodset_nbatches
(self, nbatches)
config/Config.py:211
↓ 5 callersMethodset_opt_method
(self, opt_method)
config/Config.py:244
↓ 5 callersMethodset_rel_neg_rate
(self, rate)
config/Config.py:270
↓ 5 callersMethodset_result_dir
(self, result_dir)
config/Config.py:226
↓ 5 callersMethodset_save_steps
(self, save_steps)
config/Config.py:220
↓ 5 callersMethodset_test_link
(self, test_link)
config/Config.py:199
↓ 5 callersMethodset_test_triple
(self, test_triple)
config/Config.py:202
↓ 5 callersMethodset_train_model
(self, model)
config/Config.py:304
↓ 5 callersMethodset_train_times
(self, train_times)
config/Config.py:261
↓ 5 callersMethodset_valid_steps
(self, valid_steps)
config/Config.py:217
↓ 5 callersMethodset_work_threads
(self, work_threads)
config/Config.py:264
↓ 5 callersMethodtrain
(self)
config/Config.py:411
↓ 4 callersMethod_qmult
(self, s_a, x_a, y_a, z_a, s_b, x_b, y_b, z_b)
models/OctonionE.py:53
↓ 4 callersFunctionrand_max
base/Random.h:20
↓ 4 callersMethodset_lmbda_two
(self, lmbda_two)
config/Config.py:235
↓ 2 callersMethod_calc
(self, lhs, rel, rhs, forward=True)
quate.py:17
↓ 2 callersMethod_calc
(self, e_1_h, e_2_h, e_3_h, e_4_h, e_5_h, e_6_h, e_7_h, e_8_h, e_1_t, e_2_t, e_3_t, e_4_t, e_5_t
models/OctonionE.py:91
↓ 2 callersMethod_calc
(self, s_a, x_a, y_a, z_a, s_c, x_c, y_c, z_c, s_b, x_b, y_b, z_b)
models/QuatE.py:57
↓ 2 callersMethod_qstar
(self, a, b, c, d)
models/OctonionE.py:60
↓ 2 callersFunctioncorrupt
base/Corrupt.h:118
↓ 2 callersFunctioncorrupt_head
base/Corrupt.h:7
↓ 2 callersMethodquaternion_init
(self, in_features, out_features, criterion='he')
models/QuatE.py:143
↓ 2 callersFunctionrand
[a,b)
base/Random.h:28
↓ 2 callersFunctionrandd
base/Random.h:15
↓ 2 callersMethodvalid
(self, model)
config/Config.py:392
↓ 1 callersMethod_omult
(self, a_1, a_2, a_3, a_4, b_1, b_2, b_3, b_4, c_1, c_2, c_3, c_4, d_1, d_2, d_3, d_4)
models/OctonionE.py:63
↓ 1 callersMethod_onorm
(self, r_1, r_2, r_3, r_4, r_5, r_6, r_7, r_8 )
models/OctonionE.py:77
↓ 1 callersFunctioncorrupt_rel
base/Corrupt.h:72
↓ 1 callersFunctioncorrupt_tail
base/Corrupt.h:39
↓ 1 callersFunctiongetNegTest
base/Test.h:258
↓ 1 callersFunctiongetNegValid
base/Test.h:276
↓ 1 callersMethodget_parameters
(self, param_dict, mode="numpy")
config/Config.py:285
↓ 1 callersFunctionimportTrainFiles
base/Reader.h:23
↓ 1 callersMethodinit_weights
(self)
models/OctonionE.py:35
↓ 1 callersMethodinit_weights
(self)
models/QuatE.py:31
↓ 1 callersMethodlink_prediction
(self)
config/Config.py:463
↓ 1 callersMethodloss
(self, score, regul, regul2)
models/OctonionE.py:106
↓ 1 callersMethodloss
(self, score, regul, regul2)
models/QuatE.py:77
↓ 1 callersMethodpredict
(self)
models/Model.py:40
↓ 1 callersMethodsampling
(self)
config/Config.py:351
↓ 1 callersMethodsave_best_checkpoint
(self, best_model)
config/Config.py:368
↓ 1 callersMethodsave_checkpoint
(self, model, epoch)
config/Config.py:362
↓ 1 callersMethodsave_embedding_matrix
(self, best_model)
config/Config.py:298
↓ 1 callersMethodset_test_model
(self, model, path=None)
config/Config.py:340
↓ 1 callersMethodtest
(self)
config/Config.py:522
↓ 1 callersMethodtrain_one_step
(self)
config/Config.py:372
↓ 1 callersMethodtriple_classification
(self)
config/Config.py:481
Method__init__
( self, sizes: Tuple[int, int, int], rank: int, init_size: float = 1e-3 )
quate.py:2
Method__init__
(self, config)
models/Model.py:9
Method__init__
(self, config)
models/OctonionE.py:13
Method__init__
(self, config)
models/QuatE.py:13
Method__init__
(self)
config/Config.py:25
Method_getattr__
(self, name)
config/Config.py:16
Methodcmp_head
base/Triple.h:18
Methodcmp_list
base/Triple.h:14
Methodcmp_rel
base/Triple.h:26
Methodcmp_rel2
base/Triple.h:30
Methodcmp_tail
base/Triple.h:22
Methodforward
(self, x)
quate.py:49
Methodforward
(self)
models/Model.py:37
Methodforward
(self)
models/OctonionE.py:111
Methodforward
(self)
models/QuatE.py:83
FunctiongetBatch
base/Base.cpp:60
FunctiongetBestThreshold
base/Test.h:319
FunctiongetEntityTotal
base/Setting.h:58
FunctiongetHeadBatch
base/Test.h:28
FunctiongetRelationTotal
base/Setting.h:63
FunctiongetTailBatch
base/Test.h:37
FunctiongetTestBatch
base/Test.h:293
FunctiongetTestTotal
base/Setting.h:78
FunctiongetTrainTotal
base/Setting.h:73
FunctiongetTripleTotal
base/Setting.h:68
FunctiongetValidBatch
base/Test.h:306
FunctiongetValidHeadBatch
base/Valid.h:22
FunctiongetValidHit10
base/Valid.h:81
FunctiongetValidTailBatch
base/Valid.h:31
FunctiongetValidTotal
base/Setting.h:83
FunctiongetWorkThreads
base/Setting.h:42
Methodget_negative_score
(self, score)
models/Model.py:32
Methodget_positive_score
(self, score)
models/Model.py:29
Methodget_queries
(self, queries: torch.Tensor)
quate.py:73
Methodget_rhs
(self, chunk_begin: int, chunk_size: int)
quate.py:68
FunctionimportTestFiles
base/Reader.h:128
FunctionimportTypeFiles
base/Reader.h:212
FunctioninitTest
base/Test.h:18
Functionmain
base/Base.cpp:134
Methodminimal
base/Triple.h:9
next →1–100 of 124, ranked by callers