
nefrikinoso is a machine learning project focused on predicting Chronic Kidney Disease (CKD). It utilizes various machine learning models to provide accurate CKD predictions and includes tools for model evaluation and user interaction.
- Preprocesses and prepares the CKD dataset.
- Trains and evaluates model performance using relevant metrics.
- Offers a web interface for user interaction and predictions.
- Provides an API endpoint for making predictions programmatically.
- Generates visualizations for model evaluation and feature analysis.
- Includes the following implementations:
- Novel 🤖 Voting Ensemble
- Novel 🤖 Stacked Ensemble Learning
- XGBoost
- SVM
- Decision Tree
- Logistic Regression
- K-Nearest Neighbors (KNN)
- Naive Bayes
- Random Forest
- Gradient Boosting
- CatBoost
- Neural Network
docker run netherquark/nefrikinoso
- Clone the repository.
- Navigate to the project directory.
- Install the required dependencies using
pip install -r requirements.txt
.
Run the models and generate visualisations using python main.py
.
Execute the app.py
script to launch the web interface for CKD prediction.
The API endpoint /api/predict
accepts patient data for CKD prediction.
Run the main.py
script to evaluate and compare the performance of the implemented machine learning models.
- Python 3.11
- pandas
- matplotlib
- joblib
- seaborn
- scikit-learn
- CatBoost
- XGBoost
- Flask
- GUnicorn
This project is licensed under the GNU GPLv3 License. Refer to LICENSE for more details.