__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 | |