Skip to content

v0.9.95
Compare
Choose a tag to compare
@KevinMusgrave KevinMusgrave released this 11 Dec 05:13
· 761 commits to master since this release
v0.9.95
f14a850

New

BatchEasyHardMiner

This new miner is an implementation of Improved Embeddings with Easy Positive Triplet Mining. See the documentation. Thanks @marijnl!

New metric added to AccuracyCalculator

The new metric is mean_average_precision, which is the commonly used k-nn based mAP in information retrieval.
Note that this differs from the already existing metric, mean_average_precision_at_r.

Bug fixes

  • dtype casting in MultiSimilarityMiner changed to work with autocast. See #233 by @thinline72
  • Added logic for dealing with zero rows in the weight matrix in DistanceWeightedMiner by ignoring them. For example, if the entire weight matrix is 0, then no triplets will be returned. Previously, the zero rows would cause a RuntimeError. See #230 by @tpanum
Assets 2
Loading