fix feature count and feature filtering for retraining