Small tweaks

1 files changed, 10 insertions, 7 deletions

diff --git a/extract.sh b/extract.sh
index 60e01f9..1e9443d 100755
--- a/extract.sh
+++ b/extract.sh
@@ -17,6 +17,7 @@ extract() {
 				echo Extracing games of size "$size"...
 				sqlite3 "$db_file" "$(query $1 $size)" | shuf > "playtak-$size"
 		done
+		cd ..
 }
 
 process() {
@@ -27,16 +28,19 @@ process() {
 				tail -n21 "data/check-$i"
 				echo Stripping overflows and illegal games...
 				grep -Fvxf "data/check-$i" "data/playtak-$i" > "data/good-playtak-$i"
-				echo -n "Generating training data... "
-				./pptdb "$i" "data/good-playtak-$i" generate
-				echo "Shuffling data..."
-				shuf "data/training-$i.csv" > "data/shuf-$i.csv"
-				mv "data/shuf-$i.csv" "data/training-$i.csv"
+				# echo -n "Generating training data... "
+				# ./pptdb "$i" "data/good-playtak-$i" generate
+				# echo "Shuffling data..."
+				# shuf "data/training-$i.csv" > "data/shuf-$i.csv"
+				# mv "data/shuf-$i.csv" "data/training-$i.csv"
 				echo
 		done
 }
 
-# extract notation,result
+if [ ! -f "data/playtak-5" ]; then
+		extract notation,result
+fi
+
 if [ ! -f "pptdb" ]; then
 		echo Preparing pptdb
 		make pptdb
@@ -44,6 +48,5 @@ fi
 
 # process
 
-make pptdb
 shuf -n10000 data/good-playtak-5 > data/smalltak-5 && ./pptdb 5 data/smalltak-5 generate && \
 head -n1 data/training-5.csv > t && tail -n+2 data/training-5.csv > data/train5.csv && shuf data/train5.csv >> t && mv t data/train5.csv