Skip to content

Commit

Permalink
Oops
Browse files Browse the repository at this point in the history
  • Loading branch information
Marianne Sinka committed Apr 7, 2010
1 parent 804701d commit d0be9e9
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion anopheles_brt/brt_wrap.py
Expand Up @@ -111,6 +111,10 @@ def unpack_gbm_object(brt_results, *names):
"""
namelist = np.array(brt_results.names)
return map(lambda n: maybe_array(brt_results[np.where(namelist==n)[0][0]]), names)

def print_gbm_object(brt_results, *names):
namelist = np.array(brt_results.names)
return map(lambda n: str(brt_results[np.where(namelist==n)[0][0]]), names)

def unpack_brt_trees(brt_results, layer_names, glob_name, glob_channels):
"""
Expand Down Expand Up @@ -218,7 +222,7 @@ def write_brt_results(brt_results, species_name, result_names):
varname = sanitize_species_name(species_name)
r('save(%s, file="%s")'%(varname,os.path.join(result_dirname, 'gbm.object.r')))

results = unpack_gbm_object(brt_results, *result_names)
results = print_gbm_object(brt_results, *result_names)
for n,v in zip(result_names, results):
file(os.path.join(result_dirname, n+'.txt'),'w').write(str(v))

Expand Down

0 comments on commit d0be9e9

Please sign in to comment.