modify compute rmse

neka-nat · neka-nat · commit e148c2950d3a · 2021-03-07T18:50:28.000+09:00
diff --git a/probreg/math_utils.py b/probreg/math_utils.py
@@ -26,7 +26,7 @@ def squared_kernel_sum(x, y):
 
 
 def compute_rmse(source, target_tree):
-    return sum([target_tree.query(pt)[0] for pt in source]) / source.shape[0]
+    return sum(target_tree.query(source)[0]) / source.shape[0]
 
 
 def rbf_kernel(x, y, beta):
