This project focuses on predicting Body Mass Index (BMI) using microbiome data through various linear machine learning regression models.
The project implements machine learning models to predict BMI values using metagenomic data. It includes data preprocessing, model training, evaluation, and overfitting analysis.
- Data Processing: Cleaning and preprocessing of metagenomic data
- Model Training: Implementation of various regression models
- Evaluation: Performance assessment using metrics like R² and RMSE
- Overfitting Analysis: Detection and visualization of model overfitting
-
Clone the repository: git clone [repository-url] cd Assignment-1
-
Install required packages: micromamba env create -f environment.yml
-
Data Preprocessing:
- Located in
data/directory
- Located in
-
Model Training:
- Navigate to
notebooks/model_analysis.ipynb - Runs multiple regression models on the preprocessed data
- Evaluates model performance on validation set
- Navigate to
-
Analysis:
- Feature importance visualization
- Model performance comparison
- Overfitting detection and analysis