diff --git a/object_attribute_classifier_cached_features/freq_acc_plot.py b/object_attribute_classifier_cached_features/freq_acc_plot.py index 0e4203af4f37a46cbd089f6c4babbc95f7fcf330..8aa30422a0fecf4c5be274978b4e3ac65c0c7f06 100644 --- a/object_attribute_classifier_cached_features/freq_acc_plot.py +++ b/object_attribute_classifier_cached_features/freq_acc_plot.py @@ -3,6 +3,10 @@ import ujson import pprint from collections import namedtuple from operator import attrgetter +import numpy as np +import matplotlib +matplotlib.use('pdf') +import matplotlib.pyplot as plt import constants @@ -69,6 +73,19 @@ if __name__=='__main__': # obj_test_freq_dict[obj], # ) + freq = [] + acc = [] + for record in sorted_records: + freq.append(record.train_freq) + acc.append(record.acc) + plt.plot(freq,acc) + plot_filename = os.path.join( + constants.region_object_scores_dirname, + 'obj_acc_vs_freq.pdf') + plt.savefig(plot_filename) + + + print(len(obj_pred_data))