Added float division to get same results with python 2. and 3

svaante · Apr 28, 2017 · 93e8bf7 · 93e8bf7
1 parent 5d3f38b
commit 93e8bf7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/id3/splitter.py b/id3/splitter.py
@@ -140,7 +140,7 @@ def _info_numerical(self, x, y):
                     min_attribute_counts[SplitRecord.LESS] = n - i
                     min_attribute_counts[SplitRecord.GREATER] = i
                     min_info = tmp_info
-                    min_info_pivot = (sorted_x[i - 1] + sorted_x[i]) / 2
+                    min_info_pivot = (sorted_x[i - 1] + sorted_x[i]) / 2.0
         return CalcRecord(CalcRecord.NUM,
                           min_info * np.true_divide(1, n),
                           pivot=min_info_pivot,

diff --git a/id3/tests/test_id3.py b/id3/tests/test_id3.py
@@ -29,7 +29,7 @@ def test_info_numerical():
     record = test_splitter._info_numerical(x_numerical_col, y)
     assert_equal(record.split_type, 0)
     assert_equal(record.attribute_counts.size, 2)
-    assert_almost_equal(record.pivot, 3)
+    assert_almost_equal(record.pivot, 3.5, 2)
     assert_almost_equal(record.info, 0.95, 2)