Welcome to the Mushroom Classifier repository, a machine learning-based solution for mushroom identification built using scikit-learn models in a Jupyter notebook. This project aims to accurately classify mushrooms as edible or poisonous, leveraging data analysis and preprocessing to fit the model.
The Mushroom Classifier is a machine learning project that focuses on accurate mushroom identification. Using scikit-learn models and Python in a Jupyter notebook, this project analyzes and processes a dataset from Kaggle to train a model that can classify mushrooms as edible or poisonous. The goal is to provide a reliable and accessible tool for mushroom enthusiasts and foragers.
The dataset used in this project comes from Kaggle and contains information on various mushroom species, including features such as cap shape, cap color, gill attachment, and more. The dataset was thoroughly analyzed and preprocessed to ensure the best possible model fitting.
Dataset source: Kaggle Mushroom Classification Dataset
Follow these steps to set up and run the Mushroom Classifier on your local machine:
- Clone the repository using
git clone https://github.com/Briankim254/mushroom-classifier.git
- Navigate to the project directory using
cd mushroom-classifier
- Create a virtual environment for the project using
python3 -m venv venv
- Activate the virtual environment using
source venv/bin/activate
(for Unix-based systems) orvenv\Scripts\activate
(for Windows) - Install the required dependencies using
pip install -r requirements.txt
- Launch Jupyter notebook using
jupyter notebook
The Jupyter notebook will open in your default web browser, and you can now access and interact with the Mushroom Classifier notebook.
- Open the
Mushroom_Classifier.ipynb
file in the Jupyter notebook interface - Execute the cells in sequence to load the dataset, perform data preprocessing, and train the model
- Examine the model evaluation results and experiment with different scikit-learn models or hyperparameters if desired
We welcome contributions to improve and expand the Mushroom Classifier project. To contribute, please follow these steps:
- Fork the repository and create a new branch for your changes
- Make your changes or additions to the project, including updates to the Jupyter notebook or dataset
- Create a pull request and wait for a review from a team member
Please ensure that your code adheres to best practices for code quality and documentation.
The Mushroom Classifier project is licensed under the MIT License. This allows for open collaboration and sharing of the project while ensuring that contributors retain ownership of their work.