!5012 modify readme for GAT

Merge pull request !5012 from zhanke/gat_readme
pull/5012/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 5c9348ab17

File diff suppressed because it is too large Load Diff

@ -16,7 +16,7 @@
if [ $# != 2 ]
then
echo "Usage: sh run_train.sh [SRC_PATH] [DATASET_NAME]"
echo "Usage: sh run_train_ascend.sh [SRC_PATH] [DATASET_NAME]"
exit 1
fi
@ -42,7 +42,7 @@ MINDRECORD_PATH=`pwd`/data_mr
rm -f $MINDRECORD_PATH/*
cd ../../utils/graph_to_mindrecord || exit
cd ../../../../utils/graph_to_mindrecord || exit
python writer.py --mindrecord_script $DATASET_NAME \
--mindrecord_file "$MINDRECORD_PATH/$DATASET_NAME" \

@ -16,7 +16,7 @@
if [ $# != 1 ]
then
echo "Usage: sh run_train.sh [DATASET_NAME]"
echo "Usage: sh run_train_ascend.sh [DATASET_NAME]"
exit 1
fi

@ -12,7 +12,18 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================
"""Preprocess data obtained for training"""
"""
Preprocess data obtained for training
Cora and Citeseer datasets are supported by our example, the original versions of these datasets are as follows:
@inproceedings{nr,
title={The Network Data Repository with Interactive Graph Analytics and Visualization},
author={Ryan A. Rossi and Nesreen K. Ahmed},
booktitle={AAAI},
url={http://networkrepository.com},
year={2015}
}
In this example, we use dataset splits provided by https://github.com/kimiyoung/planetoid (Zhilin Yang, William W. Cohen, Ruslan Salakhutdinov, [Revisiting Semi-Supervised Learning with Graph Embeddings](https://arxiv.org/abs/1603.08861), ICML 2016).
"""
import numpy as np
import mindspore.dataset as ds

Loading…
Cancel
Save