diff --git a/getdata.sh b/getdata.sh index 8c9fb3e..8347839 100644 --- a/getdata.sh +++ b/getdata.sh @@ -2,7 +2,7 @@ wget -O wine.csv https://huggingface.co/datasets/mstz/wine/raw/main/Wine_Quality_Data.csv num_rows=$(($(wc -l < wine.csv) - 1)) -echo "num_rows=$num_rows" +num_rows=$(expr "$num_rows" + 0) CUTOFF=${1:-num_rows} train_size=$((CUTOFF * 80 / 100))