Beans classification using decision tree and random forest with randomized search hyperparameter tuning

Meidya Koeshardianto, Kurniawan Eka Permana, Dhian Satria Yudha Kartika, Wahyudi Setiawan

Abstract


Dry-beans are a food with high protein. Dry-beans can be used as processed food products for emergency conditions such as famine, natural disasters, and war. Dry-beans can be used as a long-lasting product. To identify types of beans, manual work certainly requires a lot of time and effort. Therefore, creating a system that can classify beans in a computerized system is necessary. In this study, we classified beans using public data from Koklu. The data consists of sixteen features, seven classes with 13,611 rows. The data for each class of bean is unbalanced, so it is necessary to carry out a balanced dataset using random oversampling. Machine learning for classification using Decision Tree and Random Forest. Apart from that, hyperparameter tuning with randomize search for the number of trees 50, 75, 150, 200, and 300. The test results show that the Random Forest’s accuracy, precision, recall, and f1-score reach 0.9658 respectively. The best parameter number of trees is 300.

Full Text: PDF

Published: 2023-11-06

How to Cite this Article:

Meidya Koeshardianto, Kurniawan Eka Permana, Dhian Satria Yudha Kartika, Wahyudi Setiawan, Beans classification using decision tree and random forest with randomized search hyperparameter tuning, Commun. Math. Biol. Neurosci., 2023 (2023), Article ID 118

Copyright © 2023 Meidya Koeshardianto, Kurniawan Eka Permana, Dhian Satria Yudha Kartika, Wahyudi Setiawan. This is an open access article distributed under the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.

Commun. Math. Biol. Neurosci.

ISSN 2052-2541

Editorial Office: [email protected]

 

Copyright ©2024 CMBN