Code clean-up

[pmhalvor]

Check that all useful classes and methods have doc-strings/comments explaining what it's purpose is, and/or how it is used on our architectures.

• [ ] models.py
  • [ ] unique parameters for each BertHead child
  • [ ] doc-strings/ comments everywhere
  • [ ] move all finds to top of methods
  • [ ] clean config imports
• [ ] study.py
  • [ ] doc-strings
  • [ ] parameters
  • [ ] how to run
  • [ ] expected output
• [ ] plots.py
  • [ ] doc-strings
  • [ ] what logs the plots work for
  • [ ] Parameters / Returns
  • [ ] main plotters
• [ ] utils.py
  • [ ] doc-strings
  • [ ] check that all metrics are explained
  • [ ] Parameters and Returns (w/ types?)
  • [ ] sort by functionality?
  • [ ] clean config imports