diff options
Diffstat (limited to 'dist.cpp')
-rw-r--r-- | dist.cpp | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/dist.cpp b/dist.cpp new file mode 100644 index 0000000..aaa2167 --- /dev/null +++ b/dist.cpp @@ -0,0 +1,20 @@ +#include "mp.h" + +double mp::euclidean(const arma::rowvec &x1, const arma::rowvec &x2) +{ + return arma::norm(x1 - x2, 2); +} + +arma::mat mp::dist(const arma::mat &X, double (*distCalc)(const arma::rowvec &, const arma::rowvec &)) +{ + arma::uword n = X.n_rows; + arma::mat D(n, n, arma::fill::zeros); + + for (arma::uword i = 0; i < n; i++) + for (arma::uword j = 0; j < i; j++) { + D(i, j) = distCalc(X.row(i), X.row(j)); + D(j, i) = D(i, j); + } + + return D; +} |