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