Should I Really Eat That Mushroom?

Classifying edible and poisonous mushrooms with CatBoost gradient boosted decision trees

Caroline Arnold
Towards Data Science
6 min readAug 17, 2023

--

Most educational and real-world datasets contain categorical features. Today we will cover gradient boosted decision trees from the CatBoost library, which provides native support for categorical data. We will use a dataset of mushrooms that are either edible or poisonous. The mushrooms are described by categorical features such as their color, odor, and shape, and the question we want to answer is:

--

--

AI Consultant, PhD in Physics. I write about artificial intelligence, data analysis, science, and diversity. https://medium.com/visual-data-science