File size: 649 Bytes
087ac39 08bc54b 1e91764 619c3c3 797bd73 1e91764 619c3c3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
---
tags:
- transformers
- model
- sentiment analysis
- bahasa malaysia
---
# Based Model
Model = distilbert-base-uncased
Fine tuned the model from our own custom dataset
```
with training_args.strategy.scope():
model = TFDistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased")
trainer = TFTrainer(
model=model, # the instantiated 🤗 Transformers model to be trained
args=training_args, # training arguments, defined above
train_dataset=train_dataset, # training dataset
eval_dataset=test_dataset # evaluation dataset
)
trainer.train()
``` |