diff --git a/constants_crunchy.py b/constants_crunchy.py index 896561330eba6ea5128bd33e02c42897fa38307c..13c0c2321bf9474f6423ab4533c6c17320a388b6 100644 --- a/constants_crunchy.py +++ b/constants_crunchy.py @@ -128,8 +128,8 @@ region_model_accuracies_txt = os.path.join( 'model_accuracies.txt') # Object Attribute Classifier Evaluation Params -region_eval_on = 'test' # One of {'test','train_held_out'} -region_model_to_eval = region_model + '-' + '34000' +region_eval_on = 'train_subset' # One of {'test','train_held_out','train_subset'} +region_model_to_eval = region_model + '-' + '80000' region_attribute_scores_dirname = os.path.join( region_output_dir, diff --git a/object_attribute_classifier_cached_features/eval.py b/object_attribute_classifier_cached_features/eval.py index 47ae4f72597da00c96e9d535dd0818e68b3fda94..a022e2aa7fa1a8d9dda00184c24a13b98347e775 100644 --- a/object_attribute_classifier_cached_features/eval.py +++ b/object_attribute_classifier_cached_features/eval.py @@ -259,10 +259,12 @@ if __name__=='__main__': if constants.region_eval_on=='train_held_out': region_ids_json = constants.genome_train_held_out_region_ids + elif constants.region_eval_on=='train_subset': + region_ids_json = constants.genome_train_subset_region_ids elif constants.region_eval_on=='test': region_ids_json = constants.genome_test_region_ids else: - print "eval_on can only be 'test' or 'train_held_out'" + print "eval_on can only be 'test' or 'train_held_out' or 'train_subset'" raise print 'Creating batch generator...'