Update README.md
Browse files
README.md
CHANGED
|
@@ -1,16 +1,22 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
tags:
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
|
| 11 |
model-index:
|
| 12 |
-
|
| 13 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
---
|
| 15 |
|
| 16 |
# SubRoBERTa: Reddit Subreddit Classification Model
|
|
@@ -93,14 +99,6 @@ The model was trained on a dataset of posts from the 10 target subreddits. The d
|
|
| 93 |
|
| 94 |
The model was evaluated using accuracy and F1-macro scores. The best model was selected based on the F1-macro score.
|
| 95 |
|
| 96 |
-
## Environmental Impact
|
| 97 |
-
|
| 98 |
-
- **Hardware Type:** GPU
|
| 99 |
-
- **Hours used:** [Add your training time]
|
| 100 |
-
- **Cloud Provider:** [Add your cloud provider if applicable]
|
| 101 |
-
- **Compute Region:** [Add your region if applicable]
|
| 102 |
-
- **Carbon Emitted:** [Add if you have this information]
|
| 103 |
-
|
| 104 |
## Citation
|
| 105 |
|
| 106 |
If you use this model in your research, please cite:
|
|
@@ -109,13 +107,9 @@ If you use this model in your research, please cite:
|
|
| 109 |
@misc{SubRoBERTa,
|
| 110 |
author = {Marco Allanda},
|
| 111 |
title = {SubRoBERTa: Reddit Subreddit Classification Model},
|
| 112 |
-
year = {
|
| 113 |
publisher = {Hugging Face},
|
| 114 |
journal = {Hugging Face Hub},
|
| 115 |
howpublished = {\url{https://huggingface.co/marcoallanda/SubRoBERTa}}
|
| 116 |
}
|
| 117 |
-
```
|
| 118 |
-
|
| 119 |
-
## Contact
|
| 120 |
-
|
| 121 |
-
For questions or feedback, please open an issue on the [GitHub repository](https://github.com/marcoallanda/NLP-Project) or contact me through Hugging Face.
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
tags:
|
| 4 |
+
- roberta
|
| 5 |
+
- text-classification
|
| 6 |
+
- reddit
|
| 7 |
+
- nlp
|
| 8 |
+
- sequence-classification
|
| 9 |
+
- pytorch
|
| 10 |
+
- transformers
|
| 11 |
model-index:
|
| 12 |
+
- name: SubRoBERTa
|
| 13 |
+
results: []
|
| 14 |
+
language:
|
| 15 |
+
- en
|
| 16 |
+
metrics:
|
| 17 |
+
- accuracy
|
| 18 |
+
base_model:
|
| 19 |
+
- FacebookAI/roberta-base
|
| 20 |
---
|
| 21 |
|
| 22 |
# SubRoBERTa: Reddit Subreddit Classification Model
|
|
|
|
| 99 |
|
| 100 |
The model was evaluated using accuracy and F1-macro scores. The best model was selected based on the F1-macro score.
|
| 101 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 102 |
## Citation
|
| 103 |
|
| 104 |
If you use this model in your research, please cite:
|
|
|
|
| 107 |
@misc{SubRoBERTa,
|
| 108 |
author = {Marco Allanda},
|
| 109 |
title = {SubRoBERTa: Reddit Subreddit Classification Model},
|
| 110 |
+
year = {2025},
|
| 111 |
publisher = {Hugging Face},
|
| 112 |
journal = {Hugging Face Hub},
|
| 113 |
howpublished = {\url{https://huggingface.co/marcoallanda/SubRoBERTa}}
|
| 114 |
}
|
| 115 |
+
```
|
|
|
|
|
|
|
|
|
|
|
|