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