diff --git a/download.sh b/download.sh index 3054b45..5fc75ed 100755 --- a/download.sh +++ b/download.sh @@ -3,9 +3,10 @@ dataset_operation() { len1=$(cat ./all_games.csv.shuf | wc -l) len2=$(($len1/10)) len3=$(($len2*2)) - head -n $len1 all_games.csv.shuf > all_games.csv.test - head -n $len2 all_games.csv.shuf | tail -n $len1 > all_games.csv.dev - tail -n +($len2 + 1) all_games.csv.shuf > all_games.csv.train + len4=$(($len3+1)) + head -n $len2 all_games.csv.shuf > all_games.csv.test + head -n $len3 all_games.csv.shuf | tail -n $len2 > all_games.csv.dev + tail -n $len4 all_games.csv.shuf > all_games.csv.train rm all_games.csv.shuf wc -l all_games.csv.* }