added category_count so complex predictor will work with any amount of categories