fix dist calculation

francescoalemanno · Apr 4, 2022 · 83078c4 · 83078c4
1 parent a8cb9ce
commit 83078c4
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "KissSmoothing"
 uuid = "23b0397c-cd08-4270-956a-157331f0528f"
 authors = ["Francesco Alemanno <francescoalemanno710@gmail.com>"]
-version = "1.0.1"
+version = "1.0.2"
 
 [deps]
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"

diff --git a/rbf.png b/rbf.png
diff --git a/src/KissSmoothing.jl b/src/KissSmoothing.jl
@@ -85,9 +85,9 @@ function tps(r)
 end
 
 function dist(x, y)
-    mapreduce(+, x, y) do a, b
+    sqrt(mapreduce(+, x, y) do a, b
         abs2(a - b)
-    end
+    end)
 end
 
 struct RBF{G<:AbstractArray{Float64},C<:AbstractArray{Float64}}

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -50,5 +50,5 @@ end
     fn = fit_rbf(t,y,LinRange(0,2pi,20))
     pred_y = fn(t)
     error = sqrt(sum(abs2, pred_y .- y)/length(t))
-    @test error < 0.0005
+    @test error < 0.0006
 end