rttl's picture
Update README.md
1e9a54a
metadata
license: apache-2.0
datasets:
  - yelp_review_full
language:
  - en
model-index:
  - name: rttl-ai/bert-base-uncased-reviews
    results:
      - task:
          type: text classification
          name: multi_class_classification
        dataset:
          type: yelp_review_full
          name: yelp_review_full
          config: default
          split: test
        metrics:
          - type: accuracy
            value: 0.7037
            name: Accuracy

Model Details

Model Description: This model is a fine-tune checkpoint of bert-base-uncased, fine-tuned on yelp reviews.

  • Developed by: rttl-ai
  • Model Type: Text Classification
  • Language(s): English
  • License: Apache-2.0
  • Resources for more information:
  • The model was pre-trained with task-adaptive pre-training TAPT with an increased masking rate, no corruption strategy, and using WWM, following this paper
    • pre-trained on yelp reviews
    • fine-tuned on yelp reviews for multi-class text classification