FIX: doctest in feature_selection