Please comply with the ClassificationModel trait.