cartodb4/crankshaft
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

faux release for staging testing

Browse Source
This commit is contained in:
Andy Eschbacher
2018-03-22 10:19:29 -04:00
parent 37e6b4a228
commit 65be9befb1
72 changed files with 16924 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/py/crankshaft/crankshaft/segmentation/segmentation.py
View File

@@ -74,7 +74,8 @@ class Segmentation(object):
(target, features, target_mean,
feature_means) = self.clean_data(query, variable, feature_columns)
model_storage.create_model_table()
if model_name:
model_storage.create_model_table()
# find model if it exists and is specified
if model_name is not None:
@@ -90,7 +91,8 @@ class Segmentation(object):
rowid = self.data_provider.get_segmentation_data(params)
# store the model for later use
model_storage.set_model(model, model_name, feature_columns)
if model_name:
model_storage.set_model(model, model_name, feature_columns)
return zip(rowid[0]['ids'], result, accuracy_array)
def predict_segment(self, model, feature_columns, target_query,