File size: 814 Bytes
88f2665 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | ---
license: mit
library_name: pytorch
tags:
- graphconv
- cora
- node-classification
- graph-neural-network
---
# GRAPHCONV Node Classification on Cora
## Model Details
- **Architecture**: GRAPHCONV
- **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 graphconv \
--epochs 150 \
--lr_scheduler
```
|