Skip to content

Commit

Permalink
fix comments
Browse files Browse the repository at this point in the history
  • Loading branch information
Guillaume Lemaitre committed Jan 18, 2017
1 parent 66b9fa7 commit ce35545
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion sklearn/feature_selection/tests/test_mutual_info.py
Expand Up @@ -165,7 +165,8 @@ def test_mutual_info_classif_mixed():
for n_neighbors in [5, 7, 9]:
mi_nn = mutual_info_classif(X, y, discrete_features=[2],
n_neighbors=n_neighbors, random_state=0)
# Check that the continuous values have an higher MI
# Check that the continuous values have an higher MI with greater
# n_neighbors
assert_greater(mi_nn[0], mi[0])
assert_greater(mi_nn[1], mi[1])
# The n_neighbors should not have any effect on the discrete value
Expand Down

0 comments on commit ce35545

Please sign in to comment.