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

changing form of function to use query

Browse Source
This commit is contained in:
Stuart Lynn
2016-03-16 15:46:59 -04:00
parent f134a54c24
commit d3e1fca2b3
2 changed files with 19 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pg/sql/0.0.1/05_segmentation.sql
View File

@@ -30,8 +30,7 @@ $$ LANGUAGE plpythonu;
CREATE OR REPLACE FUNCTION
cdb_create_and_predict_segment (
segment_name TEXT,
table_name TEXT,
column_name TEXT,
query TEXT,
target_table TEXT,
geoid_column TEXT DEFAULT 'geoid',
census_table TEXT DEFAULT 'block_groups'
@@ -40,5 +39,5 @@ RETURNS TABLE (the_geom geometry, geoid text, prediction Numeric )
AS $$
from crankshaft import segmentation
# TODO: use named parameters or a dictionary
return segmentation.create_and_predict_segment(segment_name,table_name,column_name,geoid_column,census_table,target_table,'random_forest')
return segmentation.create_and_predict_segment(segment_name,query,geoid_column,census_table,target_table,'random_forest')
$$ LANGUAGE plpythonu;