__getitem__(self, idx) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
__init__(self, head, tail, rel, ent_total, rel_total, sampling_mode="normal", bern_flag=False, filter_flag=True, neg_ent=1, neg_rel=0) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
__len__(self) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
bern_flag (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
collate_fn(self, data) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
cross_sampling_flag (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
ent_total (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
filter_flag (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
freqRel (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
freqRel (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
get_ent_tot(self) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
get_rel_tot(self) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
get_tri_tot(self) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
h_of_r (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
h_of_tr (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
h_of_tr (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
head (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
lef_mean (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
neg_ent (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
neg_rel (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
r_of_ht (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
r_of_ht (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
rel (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
rel_total (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
rig_mean (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
sampling_mode (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
set_bern_flag(self, bern_flag) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
set_ent_neg_rate(self, rate) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
set_filter_flag(self, filter_flag) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
set_rel_neg_rate(self, rate) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
set_sampling_mode(self, sampling_mode) (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
t_of_hr (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
t_of_hr (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
t_of_r (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
tail (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |
tri_total (defined in OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset) | OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataset | |