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(
'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,
......
......@@ -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...'
......
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