Training section now works

2016-03-10 12:50:50 -05:00
parent f885cc9f7b
commit fcf57289fc
4 changed files with 70 additions and 29 deletions
--- a/pg/sql/0.0.1/05_segmentation.sql
+++ b/pg/sql/0.0.1/05_segmentation.sql
@@ -8,9 +8,9 @@ CREATE OR REPLACE FUNCTION
  )
 RETURNS NUMERIC
 AS $$
-  from crankshaft.segmentation import create_segemnt
+  from crankshaft import segmentation
  # TODO: use named parameters or a dictionary
-  return create_segment('table')
+  return segmentation.create_segment(segment_name,table_name,column_name,geoid_column,census_table,'random_forest')
 $$ LANGUAGE plpythonu;

 CREATE OR REPLACE FUNCTION