Skip to content

Fine tune a model with various metrics used in NLP. Model is trained on NLMCXR reports dataset.

Notifications You must be signed in to change notification settings

tdserapio/tdserapio-NLMCXR-Report-Generation-Metrics

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

NLMCXR Report Generation: Metrics

Fine tune a model with various metrics used in NLP. Model is trained on NLMCXR reports dataset. Metrics available:

  • BLEU score
  • CIDEr evaluation
  • METEOR evaluation
  • ROUGE evaluation
  • SPICE evaluation

NOTE: Although it most likely works, it has not been tested on > 1 epochs due to time and device constraints.

About

Fine tune a model with various metrics used in NLP. Model is trained on NLMCXR reports dataset.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published