|
|
| __init__ (self, head, tail, rel, ent_total, rel_total, sampling_mode="normal", bern_flag=False, filter_flag=True, neg_ent=1, neg_rel=0) |
| |
|
| __len__ (self) |
| |
|
| __getitem__ (self, idx) |
| |
|
| collate_fn (self, data) |
| |
|
| set_sampling_mode (self, sampling_mode) |
| |
|
| set_ent_neg_rate (self, rate) |
| |
|
| set_rel_neg_rate (self, rate) |
| |
|
| set_bern_flag (self, bern_flag) |
| |
|
| set_filter_flag (self, filter_flag) |
| |
|
| get_ent_tot (self) |
| |
|
| get_rel_tot (self) |
| |
|
| get_tri_tot (self) |
| |
|
|
| head = head |
| |
|
| tail = tail |
| |
|
| rel = rel |
| |
|
| rel_total = rel_total |
| |
|
| ent_total = ent_total |
| |
|
| tri_total = len(head) |
| |
|
str | sampling_mode = sampling_mode |
| |
|
| neg_ent = neg_ent |
| |
|
| neg_rel = neg_rel |
| |
|
| bern_flag = bern_flag |
| |
|
| filter_flag = filter_flag |
| |
|
int | cross_sampling_flag = None |
| |
|
dict | h_of_tr = {} |
| |
|
dict | t_of_hr = {} |
| |
|
dict | r_of_ht = {} |
| |
|
dict | h_of_r = {} |
| |
|
dict | t_of_r = {} |
| |
|
dict | freqRel = {} |
| |
|
dict | lef_mean = {} |
| |
|
dict | rig_mean = {} |
| |
|
list | t_of_hr = [] |
| |
|
list | h_of_tr = [] |
| |
|
list | r_of_ht = [] |
| |
|
int | freqRel = 0 |
| |
The documentation for this class was generated from the following file:
- seed_embeddings/OpenKE/data/PyTorchTrainDataLoader.py
1.12.0