From cfb58f78987f388ab0e0b380ced015993ec3af94 Mon Sep 17 00:00:00 2001 From: Stuart Lynn Date: Wed, 29 Jun 2016 14:46:03 +0000 Subject: [PATCH] removing full function calls to be compatiable with numpy 1.6.1 --- src/py/crankshaft/crankshaft/segmentation/segmentation.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/py/crankshaft/crankshaft/segmentation/segmentation.py b/src/py/crankshaft/crankshaft/segmentation/segmentation.py index 91444dd..ed61139 100644 --- a/src/py/crankshaft/crankshaft/segmentation/segmentation.py +++ b/src/py/crankshaft/crankshaft/segmentation/segmentation.py @@ -79,7 +79,8 @@ def create_and_predict_segment_agg(target, features, target_features, target_ids model, accuracy = train_model(clean_target, clean_features, model_parameters, 0.2) prediction = model.predict(target_features) - return zip(target_ids, prediction, np.full(prediction.shape, accuracy)) + accuracy_array = [accuracy]*prediction.shape[0] + return zip(target_ids, prediction, np.full(prediction.shape, accuracy_array)) @@ -102,7 +103,8 @@ def create_and_predict_segment(query, variable, target_query, model_params): model, accuracy = train_model(target, features, model_params, 0.2) cartodb_ids, result = predict_segment(model, feature_columns, target_query) - return zip(cartodb_ids, result, np.full(result.shape, accuracy)) + accuracy_array = [accuracy]*result.shape[0] + return zip(cartodb_ids, result, accuracy_array) def train_model(target, features, model_params, test_split):