diff options
Diffstat (limited to 'geom_matching/wasserstein/include')
-rw-r--r-- | geom_matching/wasserstein/include/auction_runner_gs.h | 2 | ||||
-rw-r--r-- | geom_matching/wasserstein/include/auction_runner_jac.h | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/geom_matching/wasserstein/include/auction_runner_gs.h b/geom_matching/wasserstein/include/auction_runner_gs.h index 80aa9f0..7968fa9 100644 --- a/geom_matching/wasserstein/include/auction_runner_gs.h +++ b/geom_matching/wasserstein/include/auction_runner_gs.h @@ -79,6 +79,7 @@ public: double getEpsilon() const { return epsilon; } double getWassersteinDistance(); double getWassersteinCost(); + double getRelativeError() const { return relativeError; }; static constexpr int maxIterNum { 25 }; // maximal number of iterations of epsilon-scaling private: // private data @@ -96,6 +97,7 @@ private: double weightAdjConst; double wassersteinDistance; double wassersteinCost; + double relativeError; // to get the 2 best items std::unique_ptr<AuctionOracle> oracle; #ifdef KEEP_UNASSIGNED_ORDERED diff --git a/geom_matching/wasserstein/include/auction_runner_jac.h b/geom_matching/wasserstein/include/auction_runner_jac.h index 22d42b0..524498a 100644 --- a/geom_matching/wasserstein/include/auction_runner_jac.h +++ b/geom_matching/wasserstein/include/auction_runner_jac.h @@ -52,6 +52,7 @@ public: double getEpsilon() const { return epsilon; } double getWassersteinDistance(); double getWassersteinCost(); + double getRelativeError() const { return relativeError; }; static constexpr double epsilonCommonRatio { 5 }; // next epsilon = current epsilon / epsilonCommonRatio static constexpr int maxIterNum { 25 }; // maximal number of iterations of epsilon-scaling private: @@ -69,6 +70,7 @@ private: double wassersteinDistance; double wassersteinCost; std::vector<IdxValPair> bidTable; + double relativeError; // to get the 2 best items std::unique_ptr<AuctionOracle> oracle; std::list<size_t> unassignedBidders; |