Upload README.md with huggingface_hub

README.md

@@ -0,0 +1,45 @@
+---
+license: mit
+library_name: pytorch
+tags:
+- sage
+- cora
+- node-classification
+- graph-neural-network
+---
+
+# SAGE Node Classification on Cora
+
+## Model Details
+- **Architecture**: SAGE
+- **Dataset**: Cora
+- **Task**: Node Classification
+
+## Usage
+
+```python
+from src.prediction import Predictor
+
+# Load model
+predictor = Predictor.from_checkpoint("best_model.pt")
+
+# Make predictions
+result = predictor.predict(graph_data)
+print(f"Predictions: {result.predictions}")
+print(f"Confidence: {result.confidence}")
+```
+
+## Files
+- `best_model.pt` - Trained model checkpoint
+- `hyperparameters.json` - Training configuration
+- `experiment_summary.json` - Final metrics
+
+## Training
+
+```bash
+python scripts/train.py \
+    --dataset cora \
+    --model sage \
+    --epochs 150 \
+    --lr_scheduler
+```