You can deploy this model and start earning money today!

Anormalmaniac/
citeseer_gat

Name: citeseer_gat
Availability: InStock

Views

No views yet

GAT Node Classification on Citeseer

Model Details

Architecture: GAT
Dataset: Citeseer
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 citeseer \
    --model gat \
    --epochs 150 \
    --lr_scheduler

Anormalmaniac/citeseer_gat

GAT Node Classification on Citeseer

Model Details

Usage

Files

Training

Anormalmaniac/
citeseer_gat