pascalrai's picture
Update README.md
ae753cc verified
|
raw
history blame
1.65 kB
metadata
library_name: transformers
tags: []
widget:
  - text: >-
      We are happy to discuss avenues of collaboration to create some
      exceptional content. Feel free to reach out to us to know more — 99845
      25874 (M) About Jack Media: Jack Media is a new-gen creative and
      influencer marketing company. Our content spans lifestyle, cooking,
      travel, fiction, comedy, and fashion connecting with audiences across the
      globe. Regards, Jack Media 78525 36587 (M) 
    example_title: Email 1
  - text: Call me on 9874569874
    example_title: Email 2
  - text: >-
      You can email me at [email protected] or call directly on 9999988888. The
      point of contact would be my manager Avisekh Raj
    example_title: Email 3

Overview:

The Model is fine-tuned for 3 class + "0" class.
The Dataset is custom annotated and contains 400 texts and the model was trained on the split of 0.76, 0.12, and 0.12.

The validation classification report is as follows:

Class Precision Recall f1
0 1.00 1.00 1.00
1 0.98 1.00 0.91
2 0.95 0.89 0.92
3 0.8 0.88 0.84
macro-avg 0.93 0.94 0.94

The test classification report is as follows:

Class Precision Recall f1
0 1.00 1.00 1.00
1 0.98 1.00 0.99
2 0.66 0.97 0.79
3 0.84 0.78 0.81
macro-avg 0.87 0.94 0.90

Possible future direction:

  1. Clean data to a good enough format as much as possible.
  2. Increase the data as much as possible. (Make sure to have data that is seen in real-usecase.)