sfqcyy
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

change gnn dir

This commit is contained in:
chentingting 2020-07-21 18:28:19 +08:00
parent 93323e1e2e
commit 8e8a197364
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
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.gnn.gcn.src.gcn import GCN from model_zoo.official.gnn.gcn.src.gcn import GCN
from model_zoo.gnn.gcn.src.metrics import LossAccuracyWrapper, TrainNetWrapper from model_zoo.official.gnn.gcn.src.metrics import LossAccuracyWrapper, TrainNetWrapper
from model_zoo.gnn.gcn.src.config import ConfigGCN from model_zoo.official.gnn.gcn.src.config import ConfigGCN
from model_zoo.gnn.gcn.src.dataset import get_adj_features_labels, get_mask from model_zoo.official.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'