complete the metris functions

[trackerpp.git] / include / hungarian.h

diff --git a/include/hungarian.h b/include/hungarian.h
index 64b8596..df1f354 100644 (file)
--- a/include/hungarian.h
+++ b/include/hungarian.h
@@ -12,4 +12,13 @@
 // @return the cost of the assignment
 int linear_sum_assignment(const Eigen::MatrixXi& cost_matrix, Eigen::VectorXi& row_ind, Eigen::VectorXi& col_ind);
 
+
+// Computes the consine distance between u and v
+// https://docs.scipy.org/doc/scipy/reference/generated/scipy.spatial.distance.cosine.html#scipy.spatial.distance.cosine
+double distance_cosine(const Eigen::VectorXd& u, const Eigen::VectorXd& v);
+
+// Computes  the Euclidean distance between u and v
+// https://docs.scipy.org/doc/scipy/reference/generated/scipy.spatial.distance.euclidean.html#scipy.spatial.distance.euclidean
+double distance_euclidean(const Eigen::VectorXd& u, const Eigen::VectorXd& v);
+
 #endif // _HUNGARIAN_H_