diff --git a/getdata.sh b/getdata.sh index 0cf816c..79a8e43 100644 --- a/getdata.sh +++ b/getdata.sh @@ -1,10 +1,8 @@ -#!/bin/bash -export LC_ALL=C wget -O wine.csv https://huggingface.co/datasets/mstz/wine/raw/main/Wine_Quality_Data.csv num_rows=$(($(wc -l < wine.csv) - 1)) -num_rows=$(expr "$num_rows" + 0) -CUTOFF=${1:-num_rows} + +CUTOFF=${1:-$num_rows} train_size=$((CUTOFF * 80 / 100)) test_size=$((CUTOFF * 10 / 100))