pytorch/ignite v0.2.0

New features and enhanced contrib module

on GitHub

Core

• We removed deprecated metric classes BinaryAccuracy and CategoricalAccuracy and which are replaced by Accuracy.

• Multilabel option for Accuracy, Precision, Recall metrics.

• Added other metrics:

  • ConfusionMatrix, IoU, mean IoU

• Operations on metrics: p = Precision(average=False)

  • apply PyTorch operators: mean_precision = p.mean()
  • indexing: precision_no_bg = p[1:]

• Improved our docs with more examples.

• Added FAQ section with best practices.

• Bug fixes

Now in Contributions module

• added TensorboardLogger
• added VisdomLogger
• added PolyaxonLogger
• improved ProgressBar
• New regression metrics
  • Median Absolute Error
  • Median Relative Absolute Error
  • Median Absolute Percentage Error
  • Geometric Mean Relative Absolute Error
  • Canberra Metric
  • Fractional Absolute Error
  • Wave Hedges Distance
  • Geometric Mean Absolute Error
• added new parameter scheduling classes and improved parameters:
  • PiecewiseLinear
  • LRScheduler
  • other helper methods
• added custom events support: CustomPeriodicEvent

Notebook examples

• VAE on MNIST
• CNN for text classification

Nighlty builds with pytorch-nightly as dependency

We also provide pip/conda nighlty builds with pytorch-nightly as dependency:

pip install pytorch-ignite-nightly

or

conda install -c pytorch ignite-nightly