Skip to content

Releases: HazyResearch/metal

v0.3.1

24 Oct 01:25
Compare
Choose a tag to compare
Updating version number

v0.3.0

23 Oct 06:05
Compare
Choose a tag to compare
  • Adds logging via a LogWriter class (and in contrib, a TensorBoardWriter class)
  • Refactors the ModelTuner class
  • Renames Classifier.train to Classifier.train_model to avoid overwriting the nn.Module.train function
  • Adds support for metrics requiring predicted probabilities (like ROC-AUC)
  • Better handling of EndModel construction
  • Other fixes and additions focused on EndModel training

v0.2.0

12 Oct 20:57
Compare
Choose a tag to compare
  • Adds torch.DataLoader and GPU support for both training and dev / evaluation sets
  • Other minor checks and fixes mainly around Classifier class and subclasses

v0.1.6

08 Sep 17:11
Compare
Choose a tag to compare
  • Fixes minor bugs with TaskHierarchy for multi-task LabelModel
  • Adds centered L2 reg. to LabelModel, with support for per-LF centering and weighting
  • Other minor checks and fixes