!3277 change directory of GCN and GAT from model_zoo to model_zoo/official/gnn
Merge pull request !3277 from chentingting/change_gnn_dir
This commit is contained in:
commit
93323e1e2e
Before Width: | Height: | Size: 6.3 MiB After Width: | Height: | Size: 6.3 MiB |
|
@ -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'
|
||||
|
|
Loading…
Reference in New Issue