change gnn dir

pull/3283/head
chentingting 5 years ago
parent 93323e1e2e
commit 8e8a197364

@ -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'

Loading…
Cancel
Save