Get the following assertion error for exercise 3
AssertionError Traceback (most recent call last)
in
9
10 # UNIT TESTS
—> 11 compute_information_gain_test(compute_information_gain)
~/work/public_tests.py in compute_information_gain_test(target)
103
104 result = target(X, y, node_indexes, 0)
→ 105 assert np.isclose(result, 0.019973, atol=1e-6), f"Wrong information gain. Expected {0.019973} got: {result}"
106
107 result = target(X, y, node_indexes, 1)
AssertionError: Wrong information gain. Expected 0.019973 got: 0.7505595942815911