Script to extract features based on feature importance