ENH: fixed MemoryError in _univariate_feature_screening(...) function