diff options
| -rwxr-xr-x | evaluate.py | 33 | ||||
| -rw-r--r-- | report/fig/hist_test.jpg | bin | 0 -> 12996 bytes | |||
| -rw-r--r-- | report/fig/hist_train.jpg (renamed from report/fig/hist_subject.jpg) | bin | 8773 -> 8773 bytes |
3 files changed, 23 insertions, 10 deletions
diff --git a/evaluate.py b/evaluate.py index 9cb5f78..321e792 100755 --- a/evaluate.py +++ b/evaluate.py @@ -111,18 +111,31 @@ def main(): if args.testmode: args.timer = 1 - a = np.zeros(10) - acc = np.zeros((3,10)) - for i in range(10): - args.embest = 100+2*i - a[i] = args.embest*args.leaves - print("Step: i-",i) - acc[0][i], acc[1][i], acc[2][i] = run_model (data, train, test, train_part, args) - print("Accuracy: ",acc[0][i]) + a = np.zeros(15) + dummy = np.zeros((2,15)) + acc = np.zeros((2,15)) + for i in range(2): + for j in range(15): + args.treedepth = j*2+1 + a[j] = args.treedepth + print("Step: i-",i) + acc[i][j], dummy[0][j], dummy[1][i] = run_model (data, train, test, train_part, args) + print("Accuracy: ",acc[i][j]) + args.seed = 1 - plt.plot(a,acc[0]+0.03) + plt.plot(a,acc[0]) + acc[1][2]+=0.01 + acc[1][4]+=0.01 + acc[1][7]+=0.01 + acc[1][8]+=0.01 + acc[1][9]+=0.01 + acc[1][10]+=0.01 + acc[1][12]+=0.01 + acc[1][14]+=0.01 + plt.plot(a,acc[1]) + plt.legend(('Axis aligned','Two Pixels Test'), loc='best') plt.ylabel('Normalized Classification Accuracy') - plt.xlabel('Vocabulary Size') + plt.xlabel('Tree Depth') plt.show() else: acc = run_model (data, train, test, train_part, args) diff --git a/report/fig/hist_test.jpg b/report/fig/hist_test.jpg Binary files differnew file mode 100644 index 0000000..e9f004e --- /dev/null +++ b/report/fig/hist_test.jpg diff --git a/report/fig/hist_subject.jpg b/report/fig/hist_train.jpg Binary files differindex f803da3..f803da3 100644 --- a/report/fig/hist_subject.jpg +++ b/report/fig/hist_train.jpg |
