You can deploy this model and start earning money today!

Anormalmaniac/
pubmed_sage

Name: pubmed_sage
Availability: InStock

Views

No views yet

SAGE Node Classification on Pubmed

Model Details

Architecture: SAGE
Dataset: Pubmed
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 pubmed \
    --model sage \
    --epochs 150 \
    --lr_scheduler

Anormalmaniac/pubmed_sage

SAGE Node Classification on Pubmed

Model Details

Usage

Files

Training

Anormalmaniac/
pubmed_sage