fix
This commit is contained in:
parent
5215d1d959
commit
840b9526b3
11
script.sh
11
script.sh
@ -1,12 +1,7 @@
|
|||||||
curl -o liver.data 'https://archive.ics.uci.edu/ml/machine-learning-databases/00225/Indian%20Liver%20Patient%20Dataset%20(ILPD).csv'
|
curl -o liver.data 'https://archive.ics.uci.edu/ml/machine-learning-databases/00225/Indian%20Liver%20Patient%20Dataset%20(ILPD).csv'
|
||||||
sort -R liver.data > liver.data.shuf
|
sort -R liver.data > liver.data.shuf
|
||||||
|
|
||||||
# Check if the CUTOFF variable is set
|
if [[ "$CUTOFF" != "0" ]]; then
|
||||||
if [ "$CUTOFF" == '0' ]; then
|
|
||||||
head -n 120 liver.data.shuf > liver.data.test
|
|
||||||
head -n 240 liver.data.shuf | tail -n 120 > liver.data.dev
|
|
||||||
tail -n +241 liver.data.shuf > liver.data.train
|
|
||||||
else
|
|
||||||
cut_train=$(( 60 * CUTOFF / 100 ))
|
cut_train=$(( 60 * CUTOFF / 100 ))
|
||||||
cut_test=$(( 20 * CUTOFF / 100 ))
|
cut_test=$(( 20 * CUTOFF / 100 ))
|
||||||
cut_dev=$(( CUTOFF - cut_train - cut_test ))
|
cut_dev=$(( CUTOFF - cut_train - cut_test ))
|
||||||
@ -14,6 +9,10 @@ else
|
|||||||
head -n $cut_train liver.data.shuf > liver.data.train
|
head -n $cut_train liver.data.shuf > liver.data.train
|
||||||
head -n $(( cut_train + cut_dev )) liver.data.shuf | tail -n $cut_dev > liver.data.dev
|
head -n $(( cut_train + cut_dev )) liver.data.shuf | tail -n $cut_dev > liver.data.dev
|
||||||
tail -n +$(( cut_train + cut_dev + 1 )) liver.data.shuf | head -n $cut_test > liver.data.test
|
tail -n +$(( cut_train + cut_dev + 1 )) liver.data.shuf | head -n $cut_test > liver.data.test
|
||||||
|
else
|
||||||
|
head -n 120 liver.data.shuf > liver.data.test
|
||||||
|
head -n 240 liver.data.shuf | tail -n 120 > liver.data.dev
|
||||||
|
tail -n +241 liver.data.shuf > liver.data.train
|
||||||
fi
|
fi
|
||||||
|
|
||||||
rm liver.data liver.data.shuf
|
rm liver.data liver.data.shuf
|
Loading…
Reference in New Issue
Block a user