| __init__(self, in_path=None, tri_file=None, ent_file=None, rel_file=None, batch_size=None, nbatches=None, threads=8, sampling_mode="normal", bern_flag=False, filter_flag=True, neg_ent=1, neg_rel=0, shuffle=True, drop_last=True) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| batch_size (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| ent_file (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| get_batch_size(self) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| get_ent_tot(self) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| get_rel_tot(self) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| get_triple_tot(self) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| in_path (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| nbatches (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| num_workers (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| rel_file (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| set_batch_size(self, batch_size) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| set_bern_flag(self, bern_flag) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| set_ent_neg_rate(self, rate) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| set_filter_flag(self, filter_flag) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| set_nbatches(self, nbatches) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| set_rel_neg_rate(self, rate) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| set_sampling_mode(self, sampling_mode) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| set_work_threads(self, work_threads) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |
| tri_file (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader | |