|
Optimizing mushroom classification through machine learning and hyperparameter tuning |
|---|---|
| รหัสดีโอไอ | |
| Creator | 1. Hamidah Maulida Khasanah 2. Afrig Aminuddin 3. Ferian Fauzi Abdulloh 4. Majid Rahardi 5. Hairani Hairani 6. Bima Pramudya |
| Title | Optimizing mushroom classification through machine learning and hyperparameter tuning |
| Publisher | Faculty of Engineering, Khon Kaen University |
| Publication Year | 2567 |
| Journal Title | Engineering and Applied Science Research |
| Journal Vol. | 51 |
| Journal No. | 5 |
| Page no. | 651-660 |
| Keyword | Machine learning, Mushroom classification, Hyperparameter tuning |
| URL Website | https://ph01.tci-thaijo.org/index.php/easr/index |
| Website title | Engineering and Applied Science Research |
| ISSN | 2539-6161 |
| Abstract | This research explores the application of machine learning in the classification of mushrooms as poisonous or edible, emphasizing the importance of optimal model performance to ensure food safety. This study compares four classification algorithms-Random Forest, Logistic Regression, Decision Tree, and Naive Bayes-before optimizing the two best models through Hyperparameter Tuning using Grid Search. The proposed method involves Exploratory Data Analysis (EDA), Data Preprocessing, Classification Modeling, Performance Evaluation, and Hyperparameter Tuning. The dataset used is Mushroom Classification data, and the results show that the Random Forest algorithm performs better with ROC values close to 100%, high recall, and good F1-Score. Hyperparameter tuning further improved the ROC and recall of the Random Forest model, emphasizing its adaptability to the nature of the dataset. This research emphasizes the importance of robust data processing and model optimization to achieve accurate and reliable predictions in mushroom classification, contributing to food safety endeavors. |
