fahrendrakhoirul commited on
Commit
fd4298f
1 Parent(s): b2fdd5e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -17,7 +17,7 @@ library_name: transformers
17
  pipeline_tag: text-classification
18
  ---
19
 
20
- **Title:** IndoBERT-EcommerceReview (v1.0)
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 = BertTokenizer.from_pretrained("fahrendrakhoirul/indobert-finetuned-ecommerce-reviews")
55
- model = AutoModelForSequenceClassification.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:
 
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: