kubapok/paranormal-or-skeptic-ISI-public
1
0
Fork 51
Code Issues Pull Requests Releases Wiki Activity

Final version classificator

Browse Source
This commit is contained in:
Wojciech Jarmosz 2021-06-22 00:22:49 +02:00
parent 91f262735e
commit 6939244314
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fine_tuning.py
View File

@ -29,7 +29,7 @@ class CustomDataset(torch.utils.data.Dataset):
return len(self.encodings["input_ids"]) return len(self.encodings["input_ids"])
data_train = list(zip(data_train_X, data_train_Y)) data_train = list(zip(data_train_X, data_train_Y))
data_train = random.sample(data_train, 5000) data_train = random.sample(data_train, 180000)
tokenizer = BertTokenizer.from_pretrained("bert-base-uncased") tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
train_X = tokenizer([text[0] for text in data_train], truncation=True, padding=True) train_X = tokenizer([text[0] for text in data_train], truncation=True, padding=True)