Update README.md
Browse files
README.md
CHANGED
|
@@ -16,7 +16,7 @@ tags:
|
|
| 16 |
* **Model Type:** PersonalityClassifier is a fine-tuned model from `google/flan-t5-xl` using annotation data for personality classification.
|
| 17 |
* **Model Date:** PersonalityClassifier was trained in Jan 2024.
|
| 18 |
* **Paper or resources for more information:** [https://arxiv.org/abs/2504.06868](https://arxiv.org/abs/2504.06868)
|
| 19 |
-
|
| 20 |
## Requirements
|
| 21 |
|
| 22 |
* `torch==2.1.0`
|
|
|
|
| 16 |
* **Model Type:** PersonalityClassifier is a fine-tuned model from `google/flan-t5-xl` using annotation data for personality classification.
|
| 17 |
* **Model Date:** PersonalityClassifier was trained in Jan 2024.
|
| 18 |
* **Paper or resources for more information:** [https://arxiv.org/abs/2504.06868](https://arxiv.org/abs/2504.06868)
|
| 19 |
+
* **Train data:** [https://huggingface.co/datasets/mirlab/personality_120000](https://huggingface.co/datasets/mirlab/personality_120000)
|
| 20 |
## Requirements
|
| 21 |
|
| 22 |
* `torch==2.1.0`
|