FIX: Remove raw_coef_ attribute that eats up memory