Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
adi271001 authored Feb 4, 2024
1 parent e94699c commit bd78fec
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions PubMed Text Classification/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ The dataset underwent preprocessing to address label sparsity and reduce the out
- Three state-of-the-art pre-trained language models were employed: BioBERT, RoBERTa, and XLNet.
- Training involved optimization for multi-label classification with a focus on improving F1 and flat validation accuracies.
- Graphs depicting training loss, F1 validation accuracy, and flat validation accuracy over epochs are available in the `results` directory.
- the training runs charts can be seen in weughts and Biases Dashboard: https://wandb.ai/ai-guru/Multi%20Label%20Classification%20of%20PubMed%20Articles%20%28Paper%20Night%20Presentation%29

## Conclusion

Expand Down

0 comments on commit bd78fec

Please sign in to comment.