diff options
author | nunzip <np.scarh@gmail.com> | 2019-02-11 17:54:38 +0000 |
---|---|---|
committer | nunzip <np.scarh@gmail.com> | 2019-02-11 17:54:38 +0000 |
commit | fdeecea750be8875113fac180abb94b54b84661e (patch) | |
tree | 196e7503833e667821df5f2c61a4992b89998d80 /evaluate.py | |
parent | 53899da2971ad3363dd26a277d3728b3b5f70594 (diff) | |
parent | fd35886dba493f3588b94bf2109877cf512663fa (diff) | |
download | e4-vision-fdeecea750be8875113fac180abb94b54b84661e.tar.gz e4-vision-fdeecea750be8875113fac180abb94b54b84661e.tar.bz2 e4-vision-fdeecea750be8875113fac180abb94b54b84661e.zip |
Merge branch 'master' of skozl.com:e4-vision
Diffstat (limited to 'evaluate.py')
-rwxr-xr-x | evaluate.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/evaluate.py b/evaluate.py index be6e940..321e792 100755 --- a/evaluate.py +++ b/evaluate.py @@ -19,7 +19,7 @@ import time parser = argparse.ArgumentParser() parser.add_argument("-d", "--data", help="Data path", action='store_true', default='data.npz') parser.add_argument("-c", "--conf_mat", help="Show visual confusion matrix", action='store_true') -parser.add_argument("-k", "--kmean", help="Perform kmean clustering with --kmean cluster centers", type=int, default=0) +parser.add_argument("-k", "--kmean", help="Perform kmean clustering with KMEAN cluster centers", type=int, default=0) parser.add_argument("-l", "--leaves", help="Maximum leaf nodes for RF classifier", type=int, default=256) parser.add_argument("-e", "--estimators", help="number of estimators to be used", type=int, default=100) parser.add_argument("-D", "--treedepth", help="depth of trees", type=int, default=5) @@ -49,6 +49,11 @@ def make_histogram(data, model, args): leaves = model.apply(data[i][j].T) leaves = np.apply_along_axis(np.bincount, axis=0, arr=leaves, minlength=args.leaves) histogram[i][j] = leaves.reshape(hist_size) + + print(histogram[0][0].shape) + plt.bar(np.arange(100), histogram[0][0].flatten()) + plt.show() + return histogram def run_model (data, train, test, train_part, args): |