Skip to content
Snippets Groups Projects
Commit 4fa768d7 authored by tgupta6's avatar tgupta6
Browse files

add train_subset in eval

parent 319b70d3
No related branches found
No related tags found
No related merge requests found
...@@ -128,8 +128,8 @@ region_model_accuracies_txt = os.path.join( ...@@ -128,8 +128,8 @@ region_model_accuracies_txt = os.path.join(
'model_accuracies.txt') 'model_accuracies.txt')
# Object Attribute Classifier Evaluation Params # Object Attribute Classifier Evaluation Params
region_eval_on = 'test' # One of {'test','train_held_out'} region_eval_on = 'train_subset' # One of {'test','train_held_out','train_subset'}
region_model_to_eval = region_model + '-' + '34000' region_model_to_eval = region_model + '-' + '80000'
region_attribute_scores_dirname = os.path.join( region_attribute_scores_dirname = os.path.join(
region_output_dir, region_output_dir,
......
...@@ -259,10 +259,12 @@ if __name__=='__main__': ...@@ -259,10 +259,12 @@ if __name__=='__main__':
if constants.region_eval_on=='train_held_out': if constants.region_eval_on=='train_held_out':
region_ids_json = constants.genome_train_held_out_region_ids 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': elif constants.region_eval_on=='test':
region_ids_json = constants.genome_test_region_ids region_ids_json = constants.genome_test_region_ids
else: 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 raise
print 'Creating batch generator...' print 'Creating batch generator...'
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment