Browse Source

change gnn path

tags/v0.6.0-beta
chentingting 5 years ago
parent
commit
f033caa7f0
19 changed files with 4 additions and 4 deletions
  1. +0
    -0
      model_zoo/official/gnn/gat/README.md
  2. +0
    -0
      model_zoo/official/gnn/gat/scripts/run_process_data.sh
  3. +0
    -0
      model_zoo/official/gnn/gat/scripts/run_train.sh
  4. +0
    -0
      model_zoo/official/gnn/gat/src/__init__.py
  5. +0
    -0
      model_zoo/official/gnn/gat/src/config.py
  6. +0
    -0
      model_zoo/official/gnn/gat/src/dataset.py
  7. +0
    -0
      model_zoo/official/gnn/gat/src/gat.py
  8. +0
    -0
      model_zoo/official/gnn/gat/src/utils.py
  9. +0
    -0
      model_zoo/official/gnn/gat/train.py
  10. +0
    -0
      model_zoo/official/gnn/gcn/README.md
  11. +0
    -0
      model_zoo/official/gnn/gcn/scripts/run_process_data.sh
  12. +0
    -0
      model_zoo/official/gnn/gcn/scripts/run_train.sh
  13. +0
    -0
      model_zoo/official/gnn/gcn/src/config.py
  14. +0
    -0
      model_zoo/official/gnn/gcn/src/dataset.py
  15. +0
    -0
      model_zoo/official/gnn/gcn/src/gcn.py
  16. +0
    -0
      model_zoo/official/gnn/gcn/src/metrics.py
  17. +0
    -0
      model_zoo/official/gnn/gcn/t-SNE_visualization_on_Cora.gif
  18. +0
    -0
      model_zoo/official/gnn/gcn/train.py
  19. +4
    -4
      tests/st/gnn/gcn/test_gcn.py

model_zoo/gat/README.md → model_zoo/official/gnn/gat/README.md View File


model_zoo/gat/scripts/run_process_data.sh → model_zoo/official/gnn/gat/scripts/run_process_data.sh View File


model_zoo/gat/scripts/run_train.sh → model_zoo/official/gnn/gat/scripts/run_train.sh View File


model_zoo/gat/src/__init__.py → model_zoo/official/gnn/gat/src/__init__.py View File


model_zoo/gat/src/config.py → model_zoo/official/gnn/gat/src/config.py View File


model_zoo/gat/src/dataset.py → model_zoo/official/gnn/gat/src/dataset.py View File


model_zoo/gat/src/gat.py → model_zoo/official/gnn/gat/src/gat.py View File


model_zoo/gat/src/utils.py → model_zoo/official/gnn/gat/src/utils.py View File


model_zoo/gat/train.py → model_zoo/official/gnn/gat/train.py View File


model_zoo/gcn/README.md → model_zoo/official/gnn/gcn/README.md View File


model_zoo/gcn/scripts/run_process_data.sh → model_zoo/official/gnn/gcn/scripts/run_process_data.sh View File


model_zoo/gcn/scripts/run_train.sh → model_zoo/official/gnn/gcn/scripts/run_train.sh View File


model_zoo/gcn/src/config.py → model_zoo/official/gnn/gcn/src/config.py View File


model_zoo/gcn/src/dataset.py → model_zoo/official/gnn/gcn/src/dataset.py View File


model_zoo/gcn/src/gcn.py → model_zoo/official/gnn/gcn/src/gcn.py View File


model_zoo/gcn/src/metrics.py → model_zoo/official/gnn/gcn/src/metrics.py View File


model_zoo/gcn/t-SNE_visualization_on_Cora.gif → model_zoo/official/gnn/gcn/t-SNE_visualization_on_Cora.gif View File


model_zoo/gcn/train.py → model_zoo/official/gnn/gcn/train.py View File


+ 4
- 4
tests/st/gnn/gcn/test_gcn.py View File

@@ -17,10 +17,10 @@ import time
import pytest import pytest
import numpy as np import numpy as np
from mindspore import context from mindspore import context
from model_zoo.gcn.src.gcn import GCN
from model_zoo.gcn.src.metrics import LossAccuracyWrapper, TrainNetWrapper
from model_zoo.gcn.src.config import ConfigGCN
from model_zoo.gcn.src.dataset import get_adj_features_labels, get_mask
from model_zoo.gnn.gcn.src.gcn import GCN
from model_zoo.gnn.gcn.src.metrics import LossAccuracyWrapper, TrainNetWrapper
from model_zoo.gnn.gcn.src.config import ConfigGCN
from model_zoo.gnn.gcn.src.dataset import get_adj_features_labels, get_mask




DATA_DIR = '/home/workspace/mindspore_dataset/cora/cora_mr/cora_mr' DATA_DIR = '/home/workspace/mindspore_dataset/cora/cora_mr/cora_mr'


Loading…
Cancel
Save