Choosing a classification trainer

Given the two types of classification, which should you choose? As stated earlier in this chapter, compared to regression models, your prediction output type will decide between binary and multi-class classification. Does your problem simply predict a value of true or false, or does it provide a more varied output based on a pre-defined value set? If your answer is the former, you need to use a binary classification. If the latter, you will need to use a multi-class classification. In this chapter, we will demonstrate both model prediction types.

For specific binary classification trainers, SDCA, LightGBM, and FastTree are the most popular options, as well as the most documented.

For specific multi-class classification trainers, LightGBM and SDCA are the most popular and best-documented options.