Update README.md
Browse files
README.md
CHANGED
@@ -162,6 +162,8 @@ This notebook establishes a baseline model for text classification on the Reuter
|
|
162 |
[Reuters Transformer Model](https://github.com/LxYuan0420/nlp/blob/main/notebooks/transformer_reuters.ipynb):
|
163 |
This notebook delves into advanced text classification using a Transformer model on the Reuters-21578 dataset. It covers the implementation details, training process, and performance metrics of using Transformer-based models for this specific task.
|
164 |
|
|
|
|
|
165 |
## Evaluation results
|
166 |
<details>
|
167 |
<summary>Transformer Model Evaluation Result</summary>
|
|
|
162 |
[Reuters Transformer Model](https://github.com/LxYuan0420/nlp/blob/main/notebooks/transformer_reuters.ipynb):
|
163 |
This notebook delves into advanced text classification using a Transformer model on the Reuters-21578 dataset. It covers the implementation details, training process, and performance metrics of using Transformer-based models for this specific task.
|
164 |
|
165 |
+
[Multilabel Stratified Sampling & Hypyerparameter Search on Reuters Dataset](https://github.com/LxYuan0420/nlp/blob/main/notebooks/transformer_reuters_hyperparameter_tuning.ipynb):
|
166 |
+
In this notebook, we explore advanced machine learning techniques through the lens of the Hugging Face Trainer API, specifically targeting Multilabel Iterative Stratified Splitting and Hyperparameter Search. The former aims to fairly distribute imbalanced datasets across multiple labels in k-fold cross-validation, maintaining a distribution closely resembling that of the complete dataset. The latter walks users through a structured hyperparameter search to fine-tune model performance for optimal results.
|
167 |
## Evaluation results
|
168 |
<details>
|
169 |
<summary>Transformer Model Evaluation Result</summary>
|