MzeeChakulaNutritionEnsembleModel / ensemble_config.json
Shakiran's picture
Upload 8 files
ab3a223 verified
raw
history blame contribute delete
640 Bytes
{
"model_name": "MzeeChakula Nutrition Ensemble",
"version": "1.0.0",
"models": [
"CRGN",
"HetGNN",
"GAT"
],
"models_config": {
"CRGN": {
"folder": "embeddings/crgn_embeddings.npy",
"weight": 0.4
},
"HetGNN": {
"folder": "embeddings/hetgnn_embeddings.npy",
"weight": 0.35
},
"GAT": {
"folder": "embeddings/gat_embeddings.npy",
"weight": 0.25
}
},
"weights": {
"CRGN": 0.4,
"HetGNN": 0.35,
"GAT": 0.25
},
"models_dir": ".",
"embedding_dim": 64,
"description": "Ensemble of CRGN, HetGNN, and GAT for elderly nutrition recommendations"
}