Fixed merge conflict

thomaskeck · Jun 5, 2017 · 5cafdef · 5cafdef
2 parents 70fd718 + 41f8023
commit 5cafdef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/PyFastBDT/utility.py b/PyFastBDT/utility.py
@@ -31,7 +31,7 @@ def flatness(probability, feature, target, weights=None, classes=[0, 1]):
         hist_inc = hist_inc.cumsum(axis=0)
         diff = (hist_n.T - hist_inc)**2
         flatness_score += diff.sum() / (100*99)
-    return flatness_score
+    return np.sqrt(flatness_score)