Hugging Face's logo

Shakiran
/
MzeeChakulaNutritionEnsembleModel

PyTorch
English
graph-neural-networks
nutrition
healthcare
recommendation-system
ensemble
Model card Files
xet
 Community
MzeeChakulaNutritionEnsembleModel
File size: 3,859 Bytes
826e920
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
---
license: apache-2.0
tags:
- graph-neural-networks
- nutrition
- healthcare
- recommendation-system
- ensemble
language:
- en
library_name: pytorch
---

# MzeeChakula Nutrition Ensemble

**Graph Neural Network Ensemble for Elderly Nutrition Planning in Uganda**

## Model Description

Production-ready ensemble combining three Graph Neural Networks:
- **CRGN (40%)** - Compositional reasoning
- **HetGNN (35%)** - Heterogeneous graph handling  
- **GAT (25%)** - Attention mechanisms

**Performance:**
- 5.36M parameters (~20.4 MB)
- ~15ms inference (CPU), ~3ms (GPU)
- 90% link prediction AUC

## Quick Start

```python
from huggingface_hub import snapshot_download
import sys

# Download model
model_path = snapshot_download(repo_id="YOUR_USERNAME/mzeechakula-nutrition")
sys.path.append(model_path)

# Load and use
from nutrition_ensemble import NutritionEnsemble

ensemble = NutritionEnsemble(model_path)
recommendations = ensemble.recommend(user_id=0, top_k=10)
print(recommendations)
```

## Installation

```bash
pip install torch numpy pandas huggingface_hub
```

## Usage

### Basic Recommendations

```python
from nutrition_ensemble import NutritionEnsemble

# Initialize
ensemble = NutritionEnsemble(".")

# Get top 10 recommendations
recs = ensemble.recommend(user_id=0, top_k=10)
print(recs)
```

### With Exclusions

```python
# Exclude already consumed foods
recs = ensemble.recommend(
    user_id=0,
    top_k=10,
    exclude_foods=[1, 2, 3]  # Food IDs to exclude
)
```

## Model Details

- **Developed by:** Nannyombi Shakiran, Yul Lam Gatkuoth
- **Institution:** Makerere University
- **Supervisor:** Ggaliwango Marvin
- **License:** MIT
- **Model type:** Graph Neural Network Ensemble
- **Task:** Food recommendation for elderly nutrition

## Training Data

- 1,048 nodes (foods, nutrients, conditions, regions)
- 14,359 edges (relationships)
- Uganda-specific nutrition knowledge graph

## Intended Use

Personalized nutrition recommendations for elderly populations in Uganda:
- Health conditions (hypertension, diabetes, anemia, frailty, etc.)
- Regional food availability (Central, Western, Eastern, Northern)
- Cultural preferences (English and Luganda support)
- Cost-conscious recommendations (~4,232 UGX per meal)

## Limitations

- Trained specifically on Ugandan food data
- Requires user health profile for optimal recommendations
- Not a substitute for professional medical advice
- Best suited for elderly population (60+ years)

## Performance Benchmarks

**CPU (Intel i7):**
- Single prediction: 0.5 ms
- Top-10 recommendations: 15 ms
- Batch (100 users): 1.5 seconds

**GPU (NVIDIA T4):**
- Single prediction: 0.1 ms
- Top-10 recommendations: 3 ms
- Batch (100 users): 0.3 seconds

## Repository Structure

```
.
├── nutrition_ensemble.py      # Main model class
├── ensemble_config.json       # Model configuration
├── requirements.txt           # Dependencies
├── embeddings/                # Pre-computed embeddings
│   ├── crgn_embeddings.npy   # 7.30 MB
│   ├── hetgnn_embeddings.npy # 0.73 MB
│   └── gat_embeddings.npy    # 1.22 MB
└── examples/
    └── quickstart.py          # Usage examples
```

## Citation

```bibtex
@software{mzeechakula2025,
  title={MzeeChakula: Graph-Enhanced Nutrition Planning for Elderly in Uganda},
  author={Nannyombi, Shakiran and Gatkuoth, Yul Lam},
  year={2025},
  institution={Makerere University},
  supervisor={Ggaliwango, Marvin},
  url={https://github.com/MzeeChakula}
}
```

## Links

- **GitHub:** https://github.com/MzeeChakula/models
- **Data Repository:** https://github.com/MzeeChakula/data
- **Issues:** https://github.com/MzeeChakula/models/issues

## Contact

- **Email:** shakirannannyombi@gmail.com
- **Collaboration:** gatkuothyullam@gmail.com

---

**Made for Uganda's elderly population** 🇺🇬