fix download.sh

This commit is contained in:
Maciej Czajka 2022-03-27 15:06:50 +02:00
parent bdde9e91e0
commit a374571c68

View File

@ -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.*
}