Update README.md
Browse files
README.md
CHANGED
|
@@ -3,8 +3,11 @@ license: apache-2.0
|
|
| 3 |
---
|
| 4 |
|
| 5 |
In this project, we have refined the capabilities of a pre-existing sentiment analysis model ("cardiffnlp/twitter-roberta-base-sentiment-latest") to assess the Big Five personality traits for a given text/sentence. By meticulously fine-tuning this model using a specially curated dataset tailored for personality traits, it has learned to correlate specific textual inputs with distinct personality characteristics. This targeted approach has significantly enhanced the model's precision in identifying the Big Five personality traits from text, outperforming other models that were developed or fine-tuned on more generalized datasets.
|
|
|
|
| 6 |
The accuracy reaches 51%, and F1 score is 42%. Both are much higher than the similar personality-detection models hosted in huggingface. In other words, our model remarkably outperforms other models.
|
|
|
|
| 7 |
Please cite: "Wang, R., and Sun, K. 2024. Personality Detection of the Characters in English Fiction" if you use this model.
|
|
|
|
| 8 |
The following provides the code to implement the task of detecting personality from an input text.
|
| 9 |
|
| 10 |
```python
|
|
|
|
| 3 |
---
|
| 4 |
|
| 5 |
In this project, we have refined the capabilities of a pre-existing sentiment analysis model ("cardiffnlp/twitter-roberta-base-sentiment-latest") to assess the Big Five personality traits for a given text/sentence. By meticulously fine-tuning this model using a specially curated dataset tailored for personality traits, it has learned to correlate specific textual inputs with distinct personality characteristics. This targeted approach has significantly enhanced the model's precision in identifying the Big Five personality traits from text, outperforming other models that were developed or fine-tuned on more generalized datasets.
|
| 6 |
+
|
| 7 |
The accuracy reaches 51%, and F1 score is 42%. Both are much higher than the similar personality-detection models hosted in huggingface. In other words, our model remarkably outperforms other models.
|
| 8 |
+
|
| 9 |
Please cite: "Wang, R., and Sun, K. 2024. Personality Detection of the Characters in English Fiction" if you use this model.
|
| 10 |
+
|
| 11 |
The following provides the code to implement the task of detecting personality from an input text.
|
| 12 |
|
| 13 |
```python
|