Add boost python, and predictor wrapper

[trackerpp.git] / src / hungarian.cpp

diff --git a/src/hungarian.cpp b/src/hungarian.cpp
index 85b49e5..7c1d8da 100644 (file)
--- a/src/hungarian.cpp
+++ b/src/hungarian.cpp
@@ -301,3 +301,15 @@ int step_six(Hungary& state)
         }
     return 4;
 }
+
+////////////////////////////////////////////////////////////////////////////////
+double distance_cosine(const VectorXd& u, const VectorXd& v)
+{
+    return (1 - u.dot(v) / std::sqrt(u.dot(u) * v.dot(v)));
+}
+
+double distance_euclidean(const VectorXd& u, const VectorXd& v)
+{
+    VectorXd d = u - v;
+    return std::sqrt(d.dot(d));
+}