-
Notifications
You must be signed in to change notification settings - Fork 0
Examples testing out BertForMaskedLM and BertForSequenceClassification. Uses dataset at https://archive.ics.uci.edu/ml/datasets/YouTube+Spam+Collection
amartyads/bert-practice-pytorch
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
These python programs were run on Spyder, to explore how BERT pretrained models are used with PyTorch. BERT youtube comments.py uses the dataset available at https://archive.ics.uci.edu/ml/datasets/YouTube+Spam+Collection , and tries to classify comments as spam or ham. BERT masked LM.py takes one sentence, sees its output in the pretrained corpus, and tries to finetune the corpus to give desired output.
About
Examples testing out BertForMaskedLM and BertForSequenceClassification. Uses dataset at https://archive.ics.uci.edu/ml/datasets/YouTube+Spam+Collection
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published