From d0be9e923b846cda8f36feab5dee46fc4bec7d75 Mon Sep 17 00:00:00 2001 From: Marianne Sinka Date: Wed, 7 Apr 2010 17:25:39 +0100 Subject: [PATCH] Oops --- anopheles_brt/brt_wrap.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/anopheles_brt/brt_wrap.py b/anopheles_brt/brt_wrap.py index 97e336a..5fcbaac 100644 --- a/anopheles_brt/brt_wrap.py +++ b/anopheles_brt/brt_wrap.py @@ -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): """ @@ -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))