|
--- |
|
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() |
|
``` |
