Over the last week or so, I've been working on making some upgrades to the superconducting state classifier model. See the first attempt here: