DeepLearningExamples/PyTorch/Recommendation/NCF/verify_dataset.sh
2019-01-23 16:59:07 +01:00

45 lines
884 B
Bash
Executable file

function get_checker {
if [[ "$OSTYPE" == "darwin"* ]]; then
checkmd5=md5
else
checkmd5=md5sum
fi
echo $checkmd5
}
function verify_1m {
# From: curl -O http://files.grouplens.org/datasets/movielens/ml-1m.zip.md5
hash=<(echo "MD5 (ml-1m.zip) = c4d9eecfca2ab87c1945afe126590906")
local checkmd5=$(get_checker)
if diff <($checkmd5 ml-1m.zip) $hash &> /dev/null
then
echo "PASSED"
else
echo "FAILED"
fi
}
function verify_20m {
# From: curl -O http://files.grouplens.org/datasets/movielens/ml-20m.zip.md5
hash=<(echo "MD5 (ml-20m.zip) = cd245b17a1ae2cc31bb14903e1204af3")
local checkmd5=$(get_checker)
if diff <($checkmd5 ml-20m.zip) $hash &> /dev/null
then
echo "PASSED"
else
echo "FAILED"
fi
}
if [[ $1 == "ml-1m" ]]
then
verify_1m
else
verify_20m
fi