EE559-Project

EE559 project of Ruiteng Li and Zhechao Zhao

Environment Configuration

• python version 3.12.11
• Run code in console under the project folder or run the environment_setup.py

  pip install -r requirements.txt

Data Exploration

• data_exploration.ipynb

Data_preprocessing (You could skip this part)

• data_preprocessing.ipynb

Using this notebook, you can generate preprocessed data.

process_mushrooms.csv is the file currently in use.

Files like X_test.csv are obsolete. The code that uses them has been commented out. However, you can still use them for testing or other purposes.

Models (Simply run the notebooks)

• Baseline Model --> baseline_model.ipynb
• Logistic Regression --> logistic_regression.ipynb
• Multi-Layer Perceptron --> mlp.ipynb
• Naive Bayes --> NaiveBayes1.1.ipynb