Notes
Installation
Datasets
How to Train
Step 1: Import Datasets
Step 2: Set Configure Parameters
Step 3: Export Results
Step 4: Train Models
How to Test
Load Models from Import Files
Read Model Parameters from Json Files
Manually Load Models
How to Get Embedding Matrix
Set Import Files
Read Parameters From Json Files
Manually Load Models Parameters
Get the Embeddings After Training
Package Reference
Config
Getting Statistics of Dataset
Setting Configuration Parameters
Setting Inpath and Outpath
Saving and Loading Models
Setting Models
Training Models
Testing Models
Models
Base Model
TransE
TransH
TransR
TransD
RESCAL
DistMult
ComplEx
OpenKE
Docs
»
Index
Index
C
|
D
|
E
|
G
|
I
|
L
|
M
|
P
|
R
|
S
|
T
C
ComplEx (class in models)
Config (class in config)
config (module)
D
DistMult (class in models)
E
export_variables() (config.Config method)
G
get_all_instance() (models.Model method)
get_all_labels() (models.Model method)
get_ent_total() (config.Config method)
get_negtive_instance() (models.Model method)
get_parameters() (config.Config method)
get_postive_instance() (models.Model method)
get_rel_total() (config.Config method)
I
import_variables() (config.Config method)
init_weights() (models.ComplEx method)
(models.DistMult method)
(models.RESCAL method)
(models.TransD method)
(models.TransE method)
(models.TransH method)
(models.TransR method)
L
loss_func() (models.ComplEx method)
(models.DistMult method)
(models.RESCAL method)
(models.TransD method)
(models.TransE method)
(models.TransH method)
(models.TransR method)
M
Model (class in models)
models (module)
P
predict() (models.ComplEx method)
(models.DistMult method)
(models.RESCAL method)
(models.TransD method)
(models.TransE method)
(models.TransH method)
(models.TransR method)
R
RESCAL (class in models)
restore_pytorch() (config.Config method)
run() (config.Config method)
S
sampling() (config.Config method)
save_parameters() (config.Config method)
save_pytorch() (config.Config method)
set_alpha() (config.Config method)
set_bern() (config.Config method)
set_dimension() (config.Config method)
set_ent_dimension() (config.Config method)
set_ent_neg_rate() (config.Config method)
set_export_files() (config.Config method)
set_export_steps() (config.Config method)
set_import_files() (config.Config method)
set_in_path() (config.Config method)
set_log_on() (config.Config method)
set_lr_decay() (config.Config method)
set_margin() (config.Config method)
set_model() (config.Config method)
set_nbatches() (config.Config method)
set_opt_method() (config.Config method)
set_out_files() (config.Config method)
set_rel_dimension() (config.Config method)
set_rel_neg_rate() (config.Config method)
set_test_flag() (config.Config method)
set_train_times() (config.Config method)
set_weight_decay() (config.Config method)
set_work_threads() (config.Config method)
T
test() (config.Config method)
TransD (class in models)
TransE (class in models)
TransH (class in models)
TransR (class in models)