Upload folder using huggingface_hub

README.md

@@ -1,3 +1,95 @@
 ---
-license: apache-2.0
+library_name: pytorch
+tags:
+- graph-neural-network
+- knowledge-graph
+- agricultural-ai
+- crop-recommendation
+- gcn
+- graph-embeddings
+license: mit
+datasets:
+- ugandan-agricultural-data
 ---
+
+# Agricultural AI Graph Embedding Models
+
+Graph neural network models trained on Ugandan agricultural knowledge graph for crop recommendation.
+
+## Model Overview
+
+This repository contains multiple graph embedding models trained on an agricultural knowledge graph with 175,318 triples representing crop-soil-climate relationships.
+
+## Models Included
+
+### Best Model: GCN (Graph Convolutional Network)
+- **File**: `best_model.pth`
+- **Accuracy**: 87.28%
+- **F1-Score**: 85.71%
+- **ROC-AUC**: 96.90%
+- **Embedding Dimension**: 100
+- **Entities**: 2,513
+- **Relations**: 15
+
+### Individual Models
+1. **GCN Model** (`gcn_model.pth`) - Best performing
+2. **TransE Model** (`transe_model.pth`) - Translation-based
+3. **DistMult Model** (`distmult_model.pth`) - Bilinear
+4. **ComplEx Model** (`complex_model.pth`) - Complex embeddings
+5. **GraphSAGE Model** (`graphsage_model.pth`) - Sampling-based
+
+## Model Metadata
+
+The `model_metadata.json` file contains:
+- Entity to ID mappings (2,513 entities)
+- Relation to ID mappings (15 relations)
+- ID to entity mappings
+- Model configuration parameters
+
+## Usage
+
+```python
+import torch
+from huggingface_hub import hf_hub_download
+
+# Download model
+model_path = hf_hub_download(
+    repo_id="Awongo/soil-crop-recommendation-model",
+    filename="best_model.pth"
+)
+
+# Download metadata
+metadata_path = hf_hub_download(
+    repo_id="Awongo/soil-crop-recommendation-model",
+    filename="model_metadata.json"
+)
+
+# Load model (pseudo-code - adjust to your model architecture)
+# model = GCNModel(num_entities=2513, num_relations=15, embedding_dim=100)
+# model.load_state_dict(torch.load(model_path, map_location='cpu'))
+# model.eval()
+```
+
+## Training Data
+
+- **Knowledge Graph**: 175,318 triples
+- **Dataset**: Ugandan agricultural data
+- **Literature**: 52 research papers
+- **Crops**: 8 major crops (maize, rice, beans, cassava, sweet potato, banana, coffee, cotton)
+
+## Application
+
+Used in production for agricultural crop recommendations based on:
+- Soil properties (pH, organic matter, nutrients)
+- Climate conditions (temperature, rainfall)
+- Knowledge graph embeddings
+
+## Citation
+
+```bibtex
+@misc{agricultural-ai-graph-models,
+  title={Agricultural AI Graph Embedding Models for Crop Recommendation},
+  year={2025},
+  publisher={Hugging Face}
+}
+```

best_model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5231df87148c0e3f25e6191392e6089af8c17679d3444f1eaaa13305dbdd19ef
+size 1175417

best_model_info.json

@@ -0,0 +1,14 @@
+{
+  "model_name": "GCN",
+  "f1_score": 0.8571428571428571,
+  "all_metrics": {
+    "accuracy": 0.87275,
+    "precision": 0.9769673704414588,
+    "recall": 0.7635,
+    "f1_score": 0.8571428571428571,
+    "roc_auc": 0.9690023749999999,
+    "average_precision": 0.9708194406937413
+  },
+  "model_path": "/content/drive/MyDrive/Final/data/processed/trained_models/best_model.pth",
+  "metadata_path": "/content/drive/MyDrive/Final/data/processed/trained_models/model_metadata.json"
+}

complex_model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e8afdc422c174cc7f25eb5dfbb3cd3603f5d0599dd69ad8ed54cb1e892bf4b56
+size 2024985

distmult_model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e192a9fdf394029f1b6efcfac1ae9ec05de8af31ae1072fcfd5c94d546394cb
+size 1013213

gcn_model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3eee81267cf9bc2cb98a96ebdf1308f740afcf6c69f4b7537efc6c72800b1f65
+size 1175405

graphsage_model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:25b629842d3038730298835a6bb76af5c0021aca6edc012de3e0d225033b767c
+size 1255477

transe_model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ba793e7be5bf89251dd1d023d76fbf18f662b6f4381ae1eff3eed89222cd6660
+size 1013197