README.md

metadata

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

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

python scripts/train.py \
    --dataset cora \
    --model sage \
    --epochs 150 \
    --lr_scheduler