-
Notifications
You must be signed in to change notification settings - Fork 3
Expand file tree
/
Copy pathREADME.txt
More file actions
36 lines (18 loc) · 978 Bytes
/
Copy pathREADME.txt
File metadata and controls
36 lines (18 loc) · 978 Bytes
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
please make sure data/ is in the same directory as all these files
- move data.zip inside of code/
- unzip data.zip
to run the model:
python OLBmain.py
what each file does:
OLBmain.py - main file, trains and tests the data
learner.py - takes in filename of the train data and creates an OLB learner object
inputimage.py - loads image
inputdata.py - loads labels and convert to json and store in memory
test.txt - test file for you to run, taken as an input to OLBmain.py train function
traindataclass.py - organizeds the train and test dataset and samples data for each different size
OLButils.py - runs pca and kmeans and unit testing the model
OLB.py - class object of olb
OLBlearner.py - learner object where all the reinforcement learning methods are located, such as q-learning, choose_action, update_qtable,etc
utils.py - another utils file for loading image
gender.txt - ground truth labels for gender class
age.txt - ground truth labels for age class