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
```