|
|
|
@ -137,33 +137,6 @@ BEGIN
|
|
|
|
|
END;
|
|
|
|
|
$$
|
|
|
|
|
LANGUAGE plpgsql VOLATILE;
|
|
|
|
|
CREATE OR REPLACE FUNCTION
|
|
|
|
|
cdb_create_segment (
|
|
|
|
|
segment_name TEXT,
|
|
|
|
|
table_name TEXT,
|
|
|
|
|
column_name TEXT,
|
|
|
|
|
geoid_column TEXT DEFAULT 'geoid',
|
|
|
|
|
census_table TEXT DEFAULT 'block_groups'
|
|
|
|
|
)
|
|
|
|
|
RETURNS NUMERIC
|
|
|
|
|
AS $$
|
|
|
|
|
from crankshaft import segmentation
|
|
|
|
|
# TODO: use named parameters or a dictionary
|
|
|
|
|
return segmentation.create_segment(segment_name,table_name,column_name,geoid_column,census_table,'random_forest')
|
|
|
|
|
$$ LANGUAGE plpythonu;
|
|
|
|
|
|
|
|
|
|
CREATE OR REPLACE FUNCTION
|
|
|
|
|
cdb_predict_segment (
|
|
|
|
|
segment_name TEXT,
|
|
|
|
|
geoid_column TEXT DEFAULT 'geoid',
|
|
|
|
|
census_table TEXT DEFAULT 'block_groups'
|
|
|
|
|
)
|
|
|
|
|
RETURNS TABLE(geoid TEXT, prediction NUMERIC)
|
|
|
|
|
AS $$
|
|
|
|
|
from crankshaft.segmentation import create_segemnt
|
|
|
|
|
# TODO: use named parameters or a dictionary
|
|
|
|
|
return create_segment('table')
|
|
|
|
|
$$ LANGUAGE plpythonu;
|
|
|
|
|
-- Make sure by default there are no permissions for publicuser
|
|
|
|
|
-- NOTE: this happens at extension creation time, as part of an implicit transaction.
|
|
|
|
|
-- REVOKE ALL PRIVILEGES ON SCHEMA cdb_crankshaft FROM PUBLIC, publicuser CASCADE;
|
|
|
|
|