From 4fa768d718ccd06b104cdeb12fa6ef137296a181 Mon Sep 17 00:00:00 2001
From: tgupta6 <tgupta6@illinois.edu>
Date: Sun, 4 Sep 2016 15:08:53 -0500
Subject: [PATCH] add train_subset in eval
---
constants_crunchy.py | 4 ++--
object_attribute_classifier_cached_features/eval.py | 4 +++-
2 files changed, 5 insertions(+), 3 deletions(-)
diff --git a/constants_crunchy.py b/constants_crunchy.py
index 8965613..13c0c23 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 47ae4f7..a022e2a 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...'
--
GitLab