diff --git a/getdata.sh b/getdata.sh index e609376..8c9fb3e 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" CUTOFF=${1:-num_rows} train_size=$((CUTOFF * 80 / 100))