cngthnh/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test

Browse Source
This commit is contained in:
thanhvc3 2024-06-19 00:05:57 +07:00
parent 936c37d0f6
commit 03f42561c6
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
data_loader.py
View File

@ -28,6 +28,8 @@ class TrainDataset(Dataset):
def __getitem__(self, idx): def __getitem__(self, idx):
ele = self.triples[idx] ele = self.triples[idx]
if (idx == 0):
print(ele)
triple, label, sub_samp = torch.LongTensor(ele['triple']), np.int32( triple, label, sub_samp = torch.LongTensor(ele['triple']), np.int32(
ele['label']), np.float32(ele['sub_samp']) ele['label']), np.float32(ele['sub_samp'])
trp_label = self.get_label(label) trp_label = self.get_label(label)