-
Notifications
You must be signed in to change notification settings - Fork 0
/
init.sh
40 lines (36 loc) · 1.22 KB
/
init.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
echo
echo 'Setting up a new virtual environment...'
echo
echo y | conda create -n rlassist python=2.7
echo
source activate rlassist
pip install --upgrade pip
pip install scipy==0.19.1
pip install psutil subprocess32 regex tensorflow==1.0.1
pip install cython
pip install unqlite
mkdir logs
mkdir data/network_inputs
mkdir data/checkpoints
echo 'done!'
echo
echo 'Downloading DeepFix dataset...'
wget https://www.cse.iitk.ac.in/users/karkare/prutor/prutor-deepfix-09-12-2017.zip -P data/
cd data
unzip prutor-deepfix-09-12-2017.zip
mv prutor-deepfix-09-12-2017/* iitk-dataset/
rm -rf prutor-deepfix-09-12-2017 prutor-deepfix-09-12-2017.zip
cd iitk-dataset/
gunzip prutor-deepfix-09-12-2017.db.gz
mv prutor-deepfix-09-12-2017.db dataset.db
cd ../..
export PYTHONPATH=.
echo 'Preprocessing DeepFix dataset...'
python data_processing/preprocess.py
echo 'Generating training and validation dataset...'
python -O data_processing/training_data_generator.py
echo 'Converting DeepFix dataset to RLAssist format...'
python -O data_processing/deepfix_to_rlassist_test_data_converter.py
echo "Make sure that your tensorflow is version 1.0.1 before proceeding, checking your tensorflow version now!"
echo
python -c 'import tensorflow as tf; print tf.__version__'