fahrendrakhoirul
commited on
Commit
•
fd4298f
1
Parent(s):
b2fdd5e
Update README.md
Browse files
README.md
CHANGED
@@ -17,7 +17,7 @@ library_name: transformers
|
|
17 |
pipeline_tag: text-classification
|
18 |
---
|
19 |
|
20 |
-
**Title:** IndoBERT-EcommerceReview
|
21 |
**Short Summary:**
|
22 |
A fine-tuned IndoBERT model for multi-label classification of customer reviews in e-commerce, focusing on product quality, customer service, and shipping/delivery.
|
23 |
**Detailed Description:**
|
@@ -51,8 +51,8 @@ class IndoBertEcommerceReview(nn.Module, PyTorchModelHubMixin):
|
|
51 |
bert = AutoModelForSequenceClassification.from_pretrained("indobenchmark/indobert-base-p1",
|
52 |
num_labels=3,
|
53 |
problem_type="multi_label_classification")
|
54 |
-
tokenizer =
|
55 |
-
model =
|
56 |
```
|
57 |
|
58 |
This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
|
|
|
17 |
pipeline_tag: text-classification
|
18 |
---
|
19 |
|
20 |
+
**Title:** IndoBERT-EcommerceReview
|
21 |
**Short Summary:**
|
22 |
A fine-tuned IndoBERT model for multi-label classification of customer reviews in e-commerce, focusing on product quality, customer service, and shipping/delivery.
|
23 |
**Detailed Description:**
|
|
|
51 |
bert = AutoModelForSequenceClassification.from_pretrained("indobenchmark/indobert-base-p1",
|
52 |
num_labels=3,
|
53 |
problem_type="multi_label_classification")
|
54 |
+
tokenizer = AutoTokenizer.from_pretrained("fahrendrakhoirul/indobert-finetuned-ecommerce-reviews")
|
55 |
+
model = IndoBertEcommerceReview.from_pretrained("fahrendrakhoirul/indobert-finetuned-ecommerce-reviews", bert=bert)
|
56 |
```
|
57 |
|
58 |
This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
|