wassname/bert_adapter
1
0
Fork 0
mirror of https://github.com/wassname/bert_adapter.git synced 2026-06-27 16:44:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Long bugfix.

Browse Source
This commit is contained in:
Anton Kiselev
2019-05-18 15:21:50 +03:00
parent cf1495d896
commit 0ffa9e170b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data.py
+2 -2
View File
@@ -52,9 +52,9 @@ class TokenizedDataFrameDataset(Dataset):
return BertInput(*[torch.LongTensor(_) for _ in [input_ids, input_mask, segment_ids]]) return BertInput(*[torch.LongTensor(_) for _ in [input_ids, input_mask, segment_ids]])
def preprocess_label(self, label: int): def preprocess_label(self, label: int):
result = torch.zeros(len(self.y_labels)) result = torch.zeros(len(self.y_labels)).long()
result[label] = 1 result[label] = 1
return torch.LongTensor(result) return result
def __getitem__(self, index) -> dict: def __getitem__(self, index) -> dict:
sample = self.df.iloc[index] sample = self.df.iloc[index]