Github user iyerr3 commented on a diff in the pull request:
https://github.com/apache/madlib/pull/295#discussion_r205562059
--- Diff:
src/ports/postgres/modules/recursive_partitioning/random_forest.sql_in ---
@@ -1557,6 +1618,32 @@ m4_ifdef(`__HAS_FUNCTION_PROPERTIES__', `CONTAINS
SQL', `');
---------------------------------------------------------------------------
---------------------------------------------------------------------------
+/**
+ Helper function to display variable importance scores (both oob and
impurity
+ importance scores for variables).
+**/
+CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.get_var_importance(
+ model_table TEXT,
+ output_table TEXT
+) RETURNS VOID AS $$
+ PythonFunction(recursive_partitioning, random_forest, get_var_importance)
+$$ LANGUAGE plpythonu VOLATILE
+m4_ifdef(`__HAS_FUNCTION_PROPERTIES__', `MODIFIES SQL DATA', `');
+
+CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.get_var_importance(
+ message TEXT
+) RETURNS TEXT AS $$
+ PythonFunction(recursive_partitioning, random_forest,
tree_importance_help_message)
--- End diff --
Good catch
---
