dmlc--dgl
yifeim
c018436588
* add lda model * tweak latent dirichlet allocation * Update README.md * Update README.md * update example index * update header * minor tweak * add example test * update doc * Update README.md * Update README.md * add partial_fit for free * Update examples/pytorch/lda/lda_model.py Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com> * Update examples/pytorch/lda/example_20newsgroups.py Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com> * Update lda_model.py * bugfix torch Gamma uses rate parameter Co-authored-by: Yifei Ma <yifeim@amazon.com> Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
46 行
855 B
Bash
46 行
855 B
Bash
#!/bin/bash
|
|
|
|
. /opt/conda/etc/profile.d/conda.sh
|
|
conda activate pytorch-ci
|
|
GCN_EXAMPLE_DIR="./examples/pytorch/"
|
|
|
|
function fail {
|
|
echo FAIL: $@
|
|
exit -1
|
|
}
|
|
|
|
function usage {
|
|
echo "Usage: $0 [cpu|gpu]"
|
|
}
|
|
|
|
# check arguments
|
|
if [ $# -ne 1 ]; then
|
|
usage
|
|
fail "Error: must specify device"
|
|
fi
|
|
|
|
if [ "$1" == "cpu" ]; then
|
|
dev=-1
|
|
elif [ "$1" == "gpu" ]; then
|
|
export CUDA_VISIBLE_DEVICES=0
|
|
dev=0
|
|
else
|
|
usage
|
|
fail "Unknown device $1"
|
|
fi
|
|
|
|
export DGLBACKEND=pytorch
|
|
export DGL_LIBRARY_PATH=${PWD}/build
|
|
export PYTHONPATH=${PWD}/python:$PYTHONPATH
|
|
export DGL_DOWNLOAD_DIR=${PWD}
|
|
|
|
# test
|
|
|
|
pushd $GCN_EXAMPLE_DIR> /dev/null
|
|
|
|
python3 pagerank.py || fail "run pagerank.py on $1"
|
|
python3 gcn/gcn.py --dataset cora --gpu $dev || fail "run gcn/gcn.py on $1"
|
|
python3 lda/lda_model.py || fail "run lda/lda_model.py on $1"
|
|
|
|
popd > /dev/null
|