diff --git a/model_zoo/gat/README.md b/model_zoo/official/gnn/gat/README.md similarity index 100% rename from model_zoo/gat/README.md rename to model_zoo/official/gnn/gat/README.md diff --git a/model_zoo/gat/scripts/run_process_data.sh b/model_zoo/official/gnn/gat/scripts/run_process_data.sh similarity index 100% rename from model_zoo/gat/scripts/run_process_data.sh rename to model_zoo/official/gnn/gat/scripts/run_process_data.sh diff --git a/model_zoo/gat/scripts/run_train.sh b/model_zoo/official/gnn/gat/scripts/run_train.sh similarity index 100% rename from model_zoo/gat/scripts/run_train.sh rename to model_zoo/official/gnn/gat/scripts/run_train.sh diff --git a/model_zoo/gat/src/__init__.py b/model_zoo/official/gnn/gat/src/__init__.py similarity index 100% rename from model_zoo/gat/src/__init__.py rename to model_zoo/official/gnn/gat/src/__init__.py diff --git a/model_zoo/gat/src/config.py b/model_zoo/official/gnn/gat/src/config.py similarity index 100% rename from model_zoo/gat/src/config.py rename to model_zoo/official/gnn/gat/src/config.py diff --git a/model_zoo/gat/src/dataset.py b/model_zoo/official/gnn/gat/src/dataset.py similarity index 100% rename from model_zoo/gat/src/dataset.py rename to model_zoo/official/gnn/gat/src/dataset.py diff --git a/model_zoo/gat/src/gat.py b/model_zoo/official/gnn/gat/src/gat.py similarity index 100% rename from model_zoo/gat/src/gat.py rename to model_zoo/official/gnn/gat/src/gat.py diff --git a/model_zoo/gat/src/utils.py b/model_zoo/official/gnn/gat/src/utils.py similarity index 100% rename from model_zoo/gat/src/utils.py rename to model_zoo/official/gnn/gat/src/utils.py diff --git a/model_zoo/gat/train.py b/model_zoo/official/gnn/gat/train.py similarity index 100% rename from model_zoo/gat/train.py rename to model_zoo/official/gnn/gat/train.py diff --git a/model_zoo/gcn/README.md b/model_zoo/official/gnn/gcn/README.md similarity index 100% rename from model_zoo/gcn/README.md rename to model_zoo/official/gnn/gcn/README.md diff --git a/model_zoo/gcn/scripts/run_process_data.sh b/model_zoo/official/gnn/gcn/scripts/run_process_data.sh similarity index 100% rename from model_zoo/gcn/scripts/run_process_data.sh rename to model_zoo/official/gnn/gcn/scripts/run_process_data.sh diff --git a/model_zoo/gcn/scripts/run_train.sh b/model_zoo/official/gnn/gcn/scripts/run_train.sh similarity index 100% rename from model_zoo/gcn/scripts/run_train.sh rename to model_zoo/official/gnn/gcn/scripts/run_train.sh diff --git a/model_zoo/gcn/src/config.py b/model_zoo/official/gnn/gcn/src/config.py similarity index 100% rename from model_zoo/gcn/src/config.py rename to model_zoo/official/gnn/gcn/src/config.py diff --git a/model_zoo/gcn/src/dataset.py b/model_zoo/official/gnn/gcn/src/dataset.py similarity index 100% rename from model_zoo/gcn/src/dataset.py rename to model_zoo/official/gnn/gcn/src/dataset.py diff --git a/model_zoo/gcn/src/gcn.py b/model_zoo/official/gnn/gcn/src/gcn.py similarity index 100% rename from model_zoo/gcn/src/gcn.py rename to model_zoo/official/gnn/gcn/src/gcn.py diff --git a/model_zoo/gcn/src/metrics.py b/model_zoo/official/gnn/gcn/src/metrics.py similarity index 100% rename from model_zoo/gcn/src/metrics.py rename to model_zoo/official/gnn/gcn/src/metrics.py diff --git a/model_zoo/gcn/t-SNE_visualization_on_Cora.gif b/model_zoo/official/gnn/gcn/t-SNE_visualization_on_Cora.gif similarity index 100% rename from model_zoo/gcn/t-SNE_visualization_on_Cora.gif rename to model_zoo/official/gnn/gcn/t-SNE_visualization_on_Cora.gif diff --git a/model_zoo/gcn/train.py b/model_zoo/official/gnn/gcn/train.py similarity index 100% rename from model_zoo/gcn/train.py rename to model_zoo/official/gnn/gcn/train.py diff --git a/tests/st/gnn/gcn/test_gcn.py b/tests/st/gnn/gcn/test_gcn.py index afe123cdb0eecd9217e782325338e4b8038557fa..8ab1f3a6572febd0fa814cbcdea6532068129860 100644 --- a/tests/st/gnn/gcn/test_gcn.py +++ b/tests/st/gnn/gcn/test_gcn.py @@ -17,10 +17,10 @@ import time import pytest import numpy as np 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'