| __init__(self, model=None, data_loader=None, train_times=1000, alpha=0.5, use_gpu=False, opt_method="sgd", save_steps=None, checkpoint_dir=None, index_dir=None, analogy_file="analogies.txt") (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| alpha (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| analogies (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| checkpoint_dir (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| data_loader (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| entity_names (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| getEntityDict(self, ent_embeddings) | OpenKE.config.Trainer.Trainer | |
| index_dir (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| load_entity_names(self, index_dir) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| lr_decay (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| model (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| opt_method (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| optimizer (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| optimizer (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| run(self, link_prediction=False, test_dataloader=None, model=None, is_analogy=False, ray=True, freq=10) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| save_steps (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| set_alpha(self, alpha) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| set_checkpoint_dir(self, checkpoint_dir) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| set_lr_decay(self, lr_decay) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| set_model(self, model) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| set_opt_method(self, opt_method) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| set_save_steps(self, save_steps, checkpoint_dir=None) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| set_train_times(self, train_times) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| set_use_gpu(self, use_gpu) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| set_weight_decay(self, weight_decay) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| to_var(self, x, use_gpu) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| train_one_step(self, data) (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| train_times (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| use_gpu (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| weight_decay (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
| work_threads (defined in OpenKE.config.Trainer.Trainer) | OpenKE.config.Trainer.Trainer | |
1.12.0