fix download.sh

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

View File

@ -1,8 +1,11 @@
dataset_operation() {
cat all_games.csv | shuf > all_games.csv.shuf
head -n 1880 all_games.csv.shuf > all_games.csv.test
head -n 3760 all_games.csv.shuf | tail -n 1880 > all_games.csv.dev
tail -n +3761 all_games.csv.shuf > all_games.csv.train
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
rm all_games.csv.shuf
wc -l all_games.csv.*
}