| <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
| <html xmlns="http://www.w3.org/1999/xhtml"> |
| <head> |
| <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> |
| <meta http-equiv="X-UA-Compatible" content="IE=9"/> |
| <meta name="generator" content="Doxygen 1.8.4"/> |
| <title>MADlib: dt.sql_in Source File</title> |
| <link href="tabs.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="jquery.js"></script> |
| <script type="text/javascript" src="dynsections.js"></script> |
| <link href="navtree.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="resize.js"></script> |
| <script type="text/javascript" src="navtree.js"></script> |
| <script type="text/javascript"> |
| $(document).ready(initResizable); |
| $(window).load(resizeHeight); |
| </script> |
| <link href="search/search.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="search/search.js"></script> |
| <script type="text/javascript"> |
| $(document).ready(function() { searchBox.OnSelectItem(0); }); |
| </script> |
| <script type="text/x-mathjax-config"> |
| MathJax.Hub.Config({ |
| extensions: ["tex2jax.js", "TeX/AMSmath.js", "TeX/AMSsymbols.js"], |
| jax: ["input/TeX","output/HTML-CSS"], |
| }); |
| </script><script src="../mathjax/MathJax.js"></script> |
| <link href="doxygen.css" rel="stylesheet" type="text/css" /> |
| </head> |
| <body> |
| <div id="top"><!-- do not remove this div, it is closed by doxygen! --> |
| <div id="titlearea"> |
| <table cellspacing="0" cellpadding="0"> |
| <tbody> |
| <tr style="height: 56px;"> |
| <td style="padding-left: 0.5em;"> |
| <div id="projectname">MADlib |
|  <span id="projectnumber">1.0</span> <span style="font-size:10pt; font-style:italic"><a href="../latest/./dt_8sql__in_source.html"> A newer version is available</a></span> |
| </div> |
| <div id="projectbrief">User Documentation</div> |
| </td> |
| </tr> |
| </tbody> |
| </table> |
| </div> |
| <!-- end header part --> |
| <!-- Generated by Doxygen 1.8.4 --> |
| <script type="text/javascript"> |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| </script> |
| <div id="navrow1" class="tabs"> |
| <ul class="tablist"> |
| <li><a href="index.html"><span>Main Page</span></a></li> |
| <li><a href="modules.html"><span>Modules</span></a></li> |
| <li> |
| <div id="MSearchBox" class="MSearchBoxInactive"> |
| <span class="left"> |
| <img id="MSearchSelect" src="search/mag_sel.png" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| alt=""/> |
| <input type="text" id="MSearchField" value="Search" accesskey="S" |
| onfocus="searchBox.OnSearchFieldFocus(true)" |
| onblur="searchBox.OnSearchFieldFocus(false)" |
| onkeyup="searchBox.OnSearchFieldChange(event)"/> |
| </span><span class="right"> |
| <a id="MSearchClose" href="javascript:searchBox.CloseResultsWindow()"><img id="MSearchCloseImg" border="0" src="search/close.png" alt=""/></a> |
| </span> |
| </div> |
| </li> |
| </ul> |
| </div> |
| </div><!-- top --> |
| <div id="side-nav" class="ui-resizable side-nav-resizable"> |
| <div id="nav-tree"> |
| <div id="nav-tree-contents"> |
| <div id="nav-sync" class="sync"></div> |
| </div> |
| </div> |
| <div id="splitbar" style="-moz-user-select:none;" |
| class="ui-resizable-handle"> |
| </div> |
| </div> |
| <script type="text/javascript"> |
| $(document).ready(function(){initNavTree('dt_8sql__in_source.html','');}); |
| </script> |
| <div id="doc-content"> |
| <!-- window showing the filter options --> |
| <div id="MSearchSelectWindow" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| <a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(0)"><span class="SelectionMark"> </span>All</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(1)"><span class="SelectionMark"> </span>Files</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(2)"><span class="SelectionMark"> </span>Functions</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(3)"><span class="SelectionMark"> </span>Groups</a></div> |
| |
| <!-- iframe showing the search results (closed by default) --> |
| <div id="MSearchResultsWindow"> |
| <iframe src="javascript:void(0)" frameborder="0" |
| name="MSearchResults" id="MSearchResults"> |
| </iframe> |
| </div> |
| |
| <div class="header"> |
| <div class="headertitle"> |
| <div class="title">dt.sql_in</div> </div> |
| </div><!--header--> |
| <div class="contents"> |
| <a href="dt_8sql__in.html">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span> <span class="comment">/* ----------------------------------------------------------------------- */</span><span class="comment">/**</span></div> |
| <div class="line"><a name="l00002"></a><span class="lineno"> 2</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00003"></a><span class="lineno"> 3</span> <span class="comment"> * @file dt.sql_in</span></div> |
| <div class="line"><a name="l00004"></a><span class="lineno"> 4</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00005"></a><span class="lineno"> 5</span> <span class="comment"> * @brief the common functions written in PL/PGSQL shared by C4.5 and RF</span></div> |
| <div class="line"><a name="l00006"></a><span class="lineno"> 6</span> <span class="comment"> * @date April 5, 2012</span></div> |
| <div class="line"><a name="l00007"></a><span class="lineno"> 7</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00008"></a><span class="lineno"> 8</span> <span class="comment"> */</span><span class="comment">/* ----------------------------------------------------------------------- */</span></div> |
| <div class="line"><a name="l00009"></a><span class="lineno"> 9</span> </div> |
| <div class="line"><a name="l00010"></a><span class="lineno"> 10</span> m4_include(`SQLCommon.m4<span class="stringliteral">')</span></div> |
| <div class="line"><a name="l00011"></a><span class="lineno"> 11</span> <span class="stringliteral"></span></div> |
| <div class="line"><a name="l00012"></a><span class="lineno"> 12</span> <span class="stringliteral">/* Own macro definitions */</span></div> |
| <div class="line"><a name="l00013"></a><span class="lineno"> 13</span> <span class="stringliteral">m4_ifelse(</span></div> |
| <div class="line"><a name="l00014"></a><span class="lineno"> 14</span> <span class="stringliteral"> m4_eval(</span></div> |
| <div class="line"><a name="l00015"></a><span class="lineno"> 15</span> <span class="stringliteral"> m4_ifdef(`__GREENPLUM__'</span>, 1, 0) &&</div> |
| <div class="line"><a name="l00016"></a><span class="lineno"> 16</span>  __DBMS_VERSION_MAJOR__ * 100 + __DBMS_VERSION_MINOR__ < 401</div> |
| <div class="line"><a name="l00017"></a><span class="lineno"> 17</span>  ), 1,</div> |
| <div class="line"><a name="l00018"></a><span class="lineno"> 18</span>  `m4_define(`__GREENPLUM_PRE_4_1__<span class="charliteral">')'</span></div> |
| <div class="line"><a name="l00019"></a><span class="lineno"> 19</span> )</div> |
| <div class="line"><a name="l00020"></a><span class="lineno"> 20</span> m4_ifelse(</div> |
| <div class="line"><a name="l00021"></a><span class="lineno"> 21</span>  m4_eval(</div> |
| <div class="line"><a name="l00022"></a><span class="lineno"> 22</span>  m4_ifdef(`__POSTGRESQL__<span class="stringliteral">', 1, 0) &&</span></div> |
| <div class="line"><a name="l00023"></a><span class="lineno"> 23</span> <span class="stringliteral"> __DBMS_VERSION_MAJOR__ < 9</span></div> |
| <div class="line"><a name="l00024"></a><span class="lineno"> 24</span> <span class="stringliteral"> ), 1,</span></div> |
| <div class="line"><a name="l00025"></a><span class="lineno"> 25</span> <span class="stringliteral"> `m4_define(`__POSTGRESQL_PRE_9_0__'</span>)<span class="stringliteral">'</span></div> |
| <div class="line"><a name="l00026"></a><span class="lineno"> 26</span> <span class="stringliteral">)</span></div> |
| <div class="line"><a name="l00027"></a><span class="lineno"> 27</span> <span class="stringliteral"></span></div> |
| <div class="line"><a name="l00028"></a><span class="lineno"> 28</span> <span class="stringliteral">m4_ifelse(</span></div> |
| <div class="line"><a name="l00029"></a><span class="lineno"> 29</span> <span class="stringliteral"> m4_eval(</span></div> |
| <div class="line"><a name="l00030"></a><span class="lineno"> 30</span> <span class="stringliteral"> m4_ifdef(`__GREENPLUM__'</span>, 1, 0) &&</div> |
| <div class="line"><a name="l00031"></a><span class="lineno"> 31</span>  __DBMS_VERSION_MAJOR__ * 10000 +</div> |
| <div class="line"><a name="l00032"></a><span class="lineno"> 32</span>  __DBMS_VERSION_MINOR__ * 100 +</div> |
| <div class="line"><a name="l00033"></a><span class="lineno"> 33</span>  __DBMS_VERSION_PATCH__ >= 40201</div> |
| <div class="line"><a name="l00034"></a><span class="lineno"> 34</span>  ), 1,</div> |
| <div class="line"><a name="l00035"></a><span class="lineno"> 35</span>  `m4_define(`__GREENPLUM_GE_4_2_1__<span class="charliteral">')'</span></div> |
| <div class="line"><a name="l00036"></a><span class="lineno"> 36</span> )</div> |
| <div class="line"><a name="l00037"></a><span class="lineno"> 37</span> </div> |
| <div class="line"><a name="l00038"></a><span class="lineno"> 38</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00039"></a><span class="lineno"> 39</span> <span class="comment"> * This is a global table to store information for various tree training.</span></div> |
| <div class="line"><a name="l00040"></a><span class="lineno"> 40</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00041"></a><span class="lineno"> 41</span> <span class="comment"> * classifier_name The name of the classifier, e.g, 'C4.5' or 'RF'.</span></div> |
| <div class="line"><a name="l00042"></a><span class="lineno"> 42</span> <span class="comment"> * result_table_oid The OID of the result table.</span></div> |
| <div class="line"><a name="l00043"></a><span class="lineno"> 43</span> <span class="comment"> * training_table_oid The OID of the training table.</span></div> |
| <div class="line"><a name="l00044"></a><span class="lineno"> 44</span> <span class="comment"> * training_metatable_oid The OID of the metadata table.</span></div> |
| <div class="line"><a name="l00045"></a><span class="lineno"> 45</span> <span class="comment"> * training_encoded_table_oid The OID of the encoded table.</span></div> |
| <div class="line"><a name="l00046"></a><span class="lineno"> 46</span> <span class="comment"> * validation_table_oid The OID of the validation table.</span></div> |
| <div class="line"><a name="l00047"></a><span class="lineno"> 47</span> <span class="comment"> * how2handle_missing_value The approach name to handle missing value.</span></div> |
| <div class="line"><a name="l00048"></a><span class="lineno"> 48</span> <span class="comment"> * split_criterion The name of the split criterion for this training.</span></div> |
| <div class="line"><a name="l00049"></a><span class="lineno"> 49</span> <span class="comment"> * sampling_percentage The sampling percentage for training each tree.</span></div> |
| <div class="line"><a name="l00050"></a><span class="lineno"> 50</span> <span class="comment"> * num_feature_chosen The number of features will be chosen to find best split.</span></div> |
| <div class="line"><a name="l00051"></a><span class="lineno"> 51</span> <span class="comment"> * num_trees The number of trees will be grow in training.</span></div> |
| <div class="line"><a name="l00052"></a><span class="lineno"> 52</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00053"></a><span class="lineno"> 53</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00054"></a><span class="lineno"> 54</span> DROP TABLE IF EXISTS MADLIB_SCHEMA.training_info;</div> |
| <div class="line"><a name="l00055"></a><span class="lineno"> 55</span> CREATE TABLE MADLIB_SCHEMA.training_info</div> |
| <div class="line"><a name="l00056"></a><span class="lineno"> 56</span>  (</div> |
| <div class="line"><a name="l00057"></a><span class="lineno"> 57</span>  classifier_name TEXT NOT NULL,</div> |
| <div class="line"><a name="l00058"></a><span class="lineno"> 58</span>  result_table_oid OID NOT NULL,</div> |
| <div class="line"><a name="l00059"></a><span class="lineno"> 59</span>  training_table_oid OID,</div> |
| <div class="line"><a name="l00060"></a><span class="lineno"> 60</span>  training_metatable_oid OID,</div> |
| <div class="line"><a name="l00061"></a><span class="lineno"> 61</span>  training_encoded_table_oid OID,</div> |
| <div class="line"><a name="l00062"></a><span class="lineno"> 62</span>  validation_table_oid OID,</div> |
| <div class="line"><a name="l00063"></a><span class="lineno"> 63</span>  how2handle_missing_value TEXT,</div> |
| <div class="line"><a name="l00064"></a><span class="lineno"> 64</span>  split_criterion TEXT,</div> |
| <div class="line"><a name="l00065"></a><span class="lineno"> 65</span>  sampling_percentage FLOAT,</div> |
| <div class="line"><a name="l00066"></a><span class="lineno"> 66</span>  num_feature_chosen INT,</div> |
| <div class="line"><a name="l00067"></a><span class="lineno"> 67</span>  num_trees INT,</div> |
| <div class="line"><a name="l00068"></a><span class="lineno"> 68</span>  PRIMARY KEY (result_table_oid)</div> |
| <div class="line"><a name="l00069"></a><span class="lineno"> 69</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (result_table_oid)');</div> |
| <div class="line"><a name="l00070"></a><span class="lineno"> 70</span> GRANT SELECT, INSERT, UPDATE, DELETE ON MADLIB_SCHEMA.training_info TO PUBLIC;</div> |
| <div class="line"><a name="l00071"></a><span class="lineno"> 71</span> </div> |
| <div class="line"><a name="l00072"></a><span class="lineno"> 72</span> </div> |
| <div class="line"><a name="l00073"></a><span class="lineno"> 73</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00074"></a><span class="lineno"> 74</span> <span class="comment"> * @brief Remove the trained tree from training info table. </span></div> |
| <div class="line"><a name="l00075"></a><span class="lineno"> 75</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00076"></a><span class="lineno"> 76</span> <span class="comment"> * @param tree_table The full name of the tree table.</span></div> |
| <div class="line"><a name="l00077"></a><span class="lineno"> 77</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00078"></a><span class="lineno"> 78</span> <span class="comment"> */</span> </div> |
| <div class="line"><a name="l00079"></a><span class="lineno"> 79</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__delete_traininginfo</div> |
| <div class="line"><a name="l00080"></a><span class="lineno"> 80</span>  (</div> |
| <div class="line"><a name="l00081"></a><span class="lineno"> 81</span>  tree_table TEXT </div> |
| <div class="line"><a name="l00082"></a><span class="lineno"> 82</span>  ) </div> |
| <div class="line"><a name="l00083"></a><span class="lineno"> 83</span> RETURNS <span class="keywordtype">void</span> AS $$</div> |
| <div class="line"><a name="l00084"></a><span class="lineno"> 84</span> BEGIN</div> |
| <div class="line"><a name="l00085"></a><span class="lineno"> 85</span>  DELETE FROM MADLIB_SCHEMA.training_info </div> |
| <div class="line"><a name="l00086"></a><span class="lineno"> 86</span>  WHERE result_table_oid = tree_table::regclass;</div> |
| <div class="line"><a name="l00087"></a><span class="lineno"> 87</span> end</div> |
| <div class="line"><a name="l00088"></a><span class="lineno"> 88</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l00089"></a><span class="lineno"> 89</span> </div> |
| <div class="line"><a name="l00090"></a><span class="lineno"> 90</span> </div> |
| <div class="line"><a name="l00091"></a><span class="lineno"> 91</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00092"></a><span class="lineno"> 92</span> <span class="comment"> * @brief Insert the trained tree into training info table. </span></div> |
| <div class="line"><a name="l00093"></a><span class="lineno"> 93</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00094"></a><span class="lineno"> 94</span> <span class="comment"> * @param classifier_table_name The name of the classifier.</span></div> |
| <div class="line"><a name="l00095"></a><span class="lineno"> 95</span> <span class="comment"> * @param result_table_name The full name of the training result table.</span></div> |
| <div class="line"><a name="l00096"></a><span class="lineno"> 96</span> <span class="comment"> * @param training_table_name The full name of the training table.</span></div> |
| <div class="line"><a name="l00097"></a><span class="lineno"> 97</span> <span class="comment"> * @param training_metatable_name The full name of metatable.</span></div> |
| <div class="line"><a name="l00098"></a><span class="lineno"> 98</span> <span class="comment"> * @param training_encoded_table_name The full name of the encoded table. </span></div> |
| <div class="line"><a name="l00099"></a><span class="lineno"> 99</span> <span class="comment"> * @param validation_table_name The full name of the validation table.</span></div> |
| <div class="line"><a name="l00100"></a><span class="lineno"> 100</span> <span class="comment"> * @param how2handle_missing_value The name of the routine to process unknown </span></div> |
| <div class="line"><a name="l00101"></a><span class="lineno"> 101</span> <span class="comment"> * values.</span></div> |
| <div class="line"><a name="l00102"></a><span class="lineno"> 102</span> <span class="comment"> * @param split_criterion The name of split criterion.</span></div> |
| <div class="line"><a name="l00103"></a><span class="lineno"> 103</span> <span class="comment"> * @param sampling_percentage The percentage of bootstrap samples size in </span></div> |
| <div class="line"><a name="l00104"></a><span class="lineno"> 104</span> <span class="comment"> * training dataset.</span></div> |
| <div class="line"><a name="l00105"></a><span class="lineno"> 105</span> <span class="comment"> * @param num_features_chosen The number of features to split on each tree</span></div> |
| <div class="line"><a name="l00106"></a><span class="lineno"> 106</span> <span class="comment"> * node. </span></div> |
| <div class="line"><a name="l00107"></a><span class="lineno"> 107</span> <span class="comment"> * @param num_trees The number of trees after completed the </span></div> |
| <div class="line"><a name="l00108"></a><span class="lineno"> 108</span> <span class="comment"> * training process.</span></div> |
| <div class="line"><a name="l00109"></a><span class="lineno"> 109</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00110"></a><span class="lineno"> 110</span> <span class="comment"> */</span> </div> |
| <div class="line"><a name="l00111"></a><span class="lineno"> 111</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__insert_into_traininginfo</div> |
| <div class="line"><a name="l00112"></a><span class="lineno"> 112</span>  (</div> |
| <div class="line"><a name="l00113"></a><span class="lineno"> 113</span>  classifier_table_name TEXT,</div> |
| <div class="line"><a name="l00114"></a><span class="lineno"> 114</span>  result_table_name TEXT,</div> |
| <div class="line"><a name="l00115"></a><span class="lineno"> 115</span>  training_table_name TEXT,</div> |
| <div class="line"><a name="l00116"></a><span class="lineno"> 116</span>  training_metatable_name TEXT,</div> |
| <div class="line"><a name="l00117"></a><span class="lineno"> 117</span>  training_encoded_table_name TEXT,</div> |
| <div class="line"><a name="l00118"></a><span class="lineno"> 118</span>  validation_table_name TEXT,</div> |
| <div class="line"><a name="l00119"></a><span class="lineno"> 119</span>  how2handle_missing_value TEXT,</div> |
| <div class="line"><a name="l00120"></a><span class="lineno"> 120</span>  split_criterion TEXT,</div> |
| <div class="line"><a name="l00121"></a><span class="lineno"> 121</span>  sampling_percentage FLOAT,</div> |
| <div class="line"><a name="l00122"></a><span class="lineno"> 122</span>  num_features_chosen INT,</div> |
| <div class="line"><a name="l00123"></a><span class="lineno"> 123</span>  num_trees INT</div> |
| <div class="line"><a name="l00124"></a><span class="lineno"> 124</span>  )</div> |
| <div class="line"><a name="l00125"></a><span class="lineno"> 125</span> RETURNS <span class="keywordtype">void</span> AS $$</div> |
| <div class="line"><a name="l00126"></a><span class="lineno"> 126</span> BEGIN</div> |
| <div class="line"><a name="l00127"></a><span class="lineno"> 127</span>  INSERT INTO MADLIB_SCHEMA.training_info VALUES</div> |
| <div class="line"><a name="l00128"></a><span class="lineno"> 128</span>  (</div> |
| <div class="line"><a name="l00129"></a><span class="lineno"> 129</span>  classifier_table_name,</div> |
| <div class="line"><a name="l00130"></a><span class="lineno"> 130</span>  result_table_name::regclass,</div> |
| <div class="line"><a name="l00131"></a><span class="lineno"> 131</span>  training_table_name::regclass,</div> |
| <div class="line"><a name="l00132"></a><span class="lineno"> 132</span>  training_metatable_name::regclass,</div> |
| <div class="line"><a name="l00133"></a><span class="lineno"> 133</span>  training_encoded_table_name::regclass,</div> |
| <div class="line"><a name="l00134"></a><span class="lineno"> 134</span>  validation_table_name::regclass,</div> |
| <div class="line"><a name="l00135"></a><span class="lineno"> 135</span>  how2handle_missing_value,</div> |
| <div class="line"><a name="l00136"></a><span class="lineno"> 136</span>  split_criterion,</div> |
| <div class="line"><a name="l00137"></a><span class="lineno"> 137</span>  sampling_percentage,</div> |
| <div class="line"><a name="l00138"></a><span class="lineno"> 138</span>  num_features_chosen,</div> |
| <div class="line"><a name="l00139"></a><span class="lineno"> 139</span>  num_trees</div> |
| <div class="line"><a name="l00140"></a><span class="lineno"> 140</span>  );</div> |
| <div class="line"><a name="l00141"></a><span class="lineno"> 141</span> END</div> |
| <div class="line"><a name="l00142"></a><span class="lineno"> 142</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l00143"></a><span class="lineno"> 143</span> </div> |
| <div class="line"><a name="l00144"></a><span class="lineno"> 144</span> </div> |
| <div class="line"><a name="l00145"></a><span class="lineno"> 145</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00146"></a><span class="lineno"> 146</span> <span class="comment"> * @brief Get the name of the encoded table. </span></div> |
| <div class="line"><a name="l00147"></a><span class="lineno"> 147</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00148"></a><span class="lineno"> 148</span> <span class="comment"> * @param tree_table The full name of the tree table.</span></div> |
| <div class="line"><a name="l00149"></a><span class="lineno"> 149</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00150"></a><span class="lineno"> 150</span> <span class="comment"> * @return The full name of the encoded table.</span></div> |
| <div class="line"><a name="l00151"></a><span class="lineno"> 151</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00152"></a><span class="lineno"> 152</span> <span class="comment"> */</span> </div> |
| <div class="line"><a name="l00153"></a><span class="lineno"> 153</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__get_encode_table_name</div> |
| <div class="line"><a name="l00154"></a><span class="lineno"> 154</span>  (</div> |
| <div class="line"><a name="l00155"></a><span class="lineno"> 155</span>  tree_table TEXT </div> |
| <div class="line"><a name="l00156"></a><span class="lineno"> 156</span>  ) </div> |
| <div class="line"><a name="l00157"></a><span class="lineno"> 157</span> RETURNS TEXT AS $$</div> |
| <div class="line"><a name="l00158"></a><span class="lineno"> 158</span> DECLARE</div> |
| <div class="line"><a name="l00159"></a><span class="lineno"> 159</span>  encoded_table_name TEXT := '';</div> |
| <div class="line"><a name="l00160"></a><span class="lineno"> 160</span> BEGIN</div> |
| <div class="line"><a name="l00161"></a><span class="lineno"> 161</span>  SELECT MADLIB_SCHEMA.__regclass_to_text(training_encoded_table_oid) </div> |
| <div class="line"><a name="l00162"></a><span class="lineno"> 162</span>  FROM MADLIB_SCHEMA.training_info</div> |
| <div class="line"><a name="l00163"></a><span class="lineno"> 163</span>  WHERE result_table_oid = tree_table::regclass </div> |
| <div class="line"><a name="l00164"></a><span class="lineno"> 164</span>  INTO encoded_table_name;</div> |
| <div class="line"><a name="l00165"></a><span class="lineno"> 165</span>  </div> |
| <div class="line"><a name="l00166"></a><span class="lineno"> 166</span>  RETURN encoded_table_name;</div> |
| <div class="line"><a name="l00167"></a><span class="lineno"> 167</span> END</div> |
| <div class="line"><a name="l00168"></a><span class="lineno"> 168</span> $$ LANGUAGE PLPGSQL STABLE;</div> |
| <div class="line"><a name="l00169"></a><span class="lineno"> 169</span> </div> |
| <div class="line"><a name="l00170"></a><span class="lineno"> 170</span> </div> |
| <div class="line"><a name="l00171"></a><span class="lineno"> 171</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00172"></a><span class="lineno"> 172</span> <span class="comment"> * @brief Test if the given table is a valid encoded one. </span></div> |
| <div class="line"><a name="l00173"></a><span class="lineno"> 173</span> <span class="comment"> * A valid encoded table has the following characteristic:</span></div> |
| <div class="line"><a name="l00174"></a><span class="lineno"> 174</span> <span class="comment"> * + Its OID is in the column "training_encoded_table_oid"</span></div> |
| <div class="line"><a name="l00175"></a><span class="lineno"> 175</span> <span class="comment"> * of training_info table.</span></div> |
| <div class="line"><a name="l00176"></a><span class="lineno"> 176</span> <span class="comment"> * + It has 5 columns, whose names are id, fid, fval,</span></div> |
| <div class="line"><a name="l00177"></a><span class="lineno"> 177</span> <span class="comment"> * is_cont and class.</span></div> |
| <div class="line"><a name="l00178"></a><span class="lineno"> 178</span> <span class="comment"> * + The types of the 5 columns are BIGINT, INT, FLOAT8</span></div> |
| <div class="line"><a name="l00179"></a><span class="lineno"> 179</span> <span class="comment"> * BOOL and INT.</span></div> |
| <div class="line"><a name="l00180"></a><span class="lineno"> 180</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00181"></a><span class="lineno"> 181</span> <span class="comment"> * @param enc_tbl_name The full name of the encoded table.</span></div> |
| <div class="line"><a name="l00182"></a><span class="lineno"> 182</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00183"></a><span class="lineno"> 183</span> <span class="comment"> * @return Ture if the given table is a valid encoded one.</span></div> |
| <div class="line"><a name="l00184"></a><span class="lineno"> 184</span> <span class="comment"> * False if it's an invalid encoded table.</span></div> |
| <div class="line"><a name="l00185"></a><span class="lineno"> 185</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00186"></a><span class="lineno"> 186</span> <span class="comment"> */</span> </div> |
| <div class="line"><a name="l00187"></a><span class="lineno"> 187</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__is_valid_enc_table</div> |
| <div class="line"><a name="l00188"></a><span class="lineno"> 188</span>  (</div> |
| <div class="line"><a name="l00189"></a><span class="lineno"> 189</span>  enc_tbl_name TEXT </div> |
| <div class="line"><a name="l00190"></a><span class="lineno"> 190</span>  ) </div> |
| <div class="line"><a name="l00191"></a><span class="lineno"> 191</span> RETURNS BOOL AS $$</div> |
| <div class="line"><a name="l00192"></a><span class="lineno"> 192</span> DECLARE</div> |
| <div class="line"><a name="l00193"></a><span class="lineno"> 193</span>  num_enc_table INT;</div> |
| <div class="line"><a name="l00194"></a><span class="lineno"> 194</span>  num_cols INT;</div> |
| <div class="line"><a name="l00195"></a><span class="lineno"> 195</span>  ret BOOL := 'f'::BOOL;</div> |
| <div class="line"><a name="l00196"></a><span class="lineno"> 196</span> BEGIN</div> |
| <div class="line"><a name="l00197"></a><span class="lineno"> 197</span>  -- test if the table is in the training_info table</div> |
| <div class="line"><a name="l00198"></a><span class="lineno"> 198</span>  SELECT count(*) </div> |
| <div class="line"><a name="l00199"></a><span class="lineno"> 199</span>  FROM MADLIB_SCHEMA.training_info</div> |
| <div class="line"><a name="l00200"></a><span class="lineno"> 200</span>  WHERE MADLIB_SCHEMA.__regclass_to_text(training_encoded_table_oid) = </div> |
| <div class="line"><a name="l00201"></a><span class="lineno"> 201</span>  enc_tbl_name</div> |
| <div class="line"><a name="l00202"></a><span class="lineno"> 202</span>  INTO num_enc_table;</div> |
| <div class="line"><a name="l00203"></a><span class="lineno"> 203</span>  </div> |
| <div class="line"><a name="l00204"></a><span class="lineno"> 204</span>  -- test if the name and the type of a column are valid or not</div> |
| <div class="line"><a name="l00205"></a><span class="lineno"> 205</span>  SELECT count(*) </div> |
| <div class="line"><a name="l00206"></a><span class="lineno"> 206</span>  FROM pg_attribute </div> |
| <div class="line"><a name="l00207"></a><span class="lineno"> 207</span>  WHERE attrelid= enc_tbl_name::regclass::oid AND</div> |
| <div class="line"><a name="l00208"></a><span class="lineno"> 208</span>  attnum > 0 AND </div> |
| <div class="line"><a name="l00209"></a><span class="lineno"> 209</span>  not attisdropped AND</div> |
| <div class="line"><a name="l00210"></a><span class="lineno"> 210</span>  attname in ('<span class="keywordtype">id</span>', 'fid', 'fval', 'is_cont', 'class') AND</div> |
| <div class="line"><a name="l00211"></a><span class="lineno"> 211</span>  atttypid in ('int8'::regtype, '<span class="keywordtype">int</span>'::regtype, 'float8'::regtype, </div> |
| <div class="line"><a name="l00212"></a><span class="lineno"> 212</span>  '<span class="keywordtype">bool</span>'::regtype, '<span class="keywordtype">int</span>'::regtype)</div> |
| <div class="line"><a name="l00213"></a><span class="lineno"> 213</span>  INTO num_cols;</div> |
| <div class="line"><a name="l00214"></a><span class="lineno"> 214</span> </div> |
| <div class="line"><a name="l00215"></a><span class="lineno"> 215</span>  IF ((num_enc_table > 0) AND (num_cols = 5)) THEN</div> |
| <div class="line"><a name="l00216"></a><span class="lineno"> 216</span>  ret = 't'::BOOL; </div> |
| <div class="line"><a name="l00217"></a><span class="lineno"> 217</span>  END IF;</div> |
| <div class="line"><a name="l00218"></a><span class="lineno"> 218</span>  </div> |
| <div class="line"><a name="l00219"></a><span class="lineno"> 219</span>  RETURN ret;</div> |
| <div class="line"><a name="l00220"></a><span class="lineno"> 220</span> END</div> |
| <div class="line"><a name="l00221"></a><span class="lineno"> 221</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l00222"></a><span class="lineno"> 222</span> </div> |
| <div class="line"><a name="l00223"></a><span class="lineno"> 223</span> </div> |
| <div class="line"><a name="l00224"></a><span class="lineno"> 224</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00225"></a><span class="lineno"> 225</span> <span class="comment"> * @brief Get the meta table name by the tree table name. </span></div> |
| <div class="line"><a name="l00226"></a><span class="lineno"> 226</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00227"></a><span class="lineno"> 227</span> <span class="comment"> * @param tree_table The full name of the tree table.</span></div> |
| <div class="line"><a name="l00228"></a><span class="lineno"> 228</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00229"></a><span class="lineno"> 229</span> <span class="comment"> * @return The full name of the metatable.</span></div> |
| <div class="line"><a name="l00230"></a><span class="lineno"> 230</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00231"></a><span class="lineno"> 231</span> <span class="comment"> */</span> </div> |
| <div class="line"><a name="l00232"></a><span class="lineno"> 232</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__get_metatable_name</div> |
| <div class="line"><a name="l00233"></a><span class="lineno"> 233</span>  (</div> |
| <div class="line"><a name="l00234"></a><span class="lineno"> 234</span>  tree_table TEXT </div> |
| <div class="line"><a name="l00235"></a><span class="lineno"> 235</span>  ) </div> |
| <div class="line"><a name="l00236"></a><span class="lineno"> 236</span> RETURNS TEXT AS $$</div> |
| <div class="line"><a name="l00237"></a><span class="lineno"> 237</span> DECLARE</div> |
| <div class="line"><a name="l00238"></a><span class="lineno"> 238</span>  metatable_name TEXT := '';</div> |
| <div class="line"><a name="l00239"></a><span class="lineno"> 239</span> BEGIN</div> |
| <div class="line"><a name="l00240"></a><span class="lineno"> 240</span> </div> |
| <div class="line"><a name="l00241"></a><span class="lineno"> 241</span>  PERFORM MADLIB_SCHEMA.__assert_table</div> |
| <div class="line"><a name="l00242"></a><span class="lineno"> 242</span>  (</div> |
| <div class="line"><a name="l00243"></a><span class="lineno"> 243</span>  tree_table::TEXT,</div> |
| <div class="line"><a name="l00244"></a><span class="lineno"> 244</span>  't'::BOOL</div> |
| <div class="line"><a name="l00245"></a><span class="lineno"> 245</span>  ); </div> |
| <div class="line"><a name="l00246"></a><span class="lineno"> 246</span>  </div> |
| <div class="line"><a name="l00247"></a><span class="lineno"> 247</span>  PERFORM MADLIB_SCHEMA.__assert_table</div> |
| <div class="line"><a name="l00248"></a><span class="lineno"> 248</span>  (</div> |
| <div class="line"><a name="l00249"></a><span class="lineno"> 249</span>  'MADLIB_SCHEMA.training_info'::TEXT,</div> |
| <div class="line"><a name="l00250"></a><span class="lineno"> 250</span>  't'::BOOL</div> |
| <div class="line"><a name="l00251"></a><span class="lineno"> 251</span>  ); </div> |
| <div class="line"><a name="l00252"></a><span class="lineno"> 252</span>  </div> |
| <div class="line"><a name="l00253"></a><span class="lineno"> 253</span>  SELECT MADLIB_SCHEMA.__regclass_to_text(training_metatable_oid) </div> |
| <div class="line"><a name="l00254"></a><span class="lineno"> 254</span>  FROM MADLIB_SCHEMA.training_info </div> |
| <div class="line"><a name="l00255"></a><span class="lineno"> 255</span>  WHERE result_table_oid = tree_table::regclass </div> |
| <div class="line"><a name="l00256"></a><span class="lineno"> 256</span>  INTO metatable_name;</div> |
| <div class="line"><a name="l00257"></a><span class="lineno"> 257</span>  </div> |
| <div class="line"><a name="l00258"></a><span class="lineno"> 258</span>  RETURN metatable_name;</div> |
| <div class="line"><a name="l00259"></a><span class="lineno"> 259</span> END</div> |
| <div class="line"><a name="l00260"></a><span class="lineno"> 260</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l00261"></a><span class="lineno"> 261</span> </div> |
| <div class="line"><a name="l00262"></a><span class="lineno"> 262</span> </div> |
| <div class="line"><a name="l00263"></a><span class="lineno"> 263</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00264"></a><span class="lineno"> 264</span> <span class="comment"> * @brief Get the unknown values processing routine id. </span></div> |
| <div class="line"><a name="l00265"></a><span class="lineno"> 265</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00266"></a><span class="lineno"> 266</span> <span class="comment"> * @param tree_table The full name of the tree table.</span></div> |
| <div class="line"><a name="l00267"></a><span class="lineno"> 267</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00268"></a><span class="lineno"> 268</span> <span class="comment"> * @return The encoded missing value processing routine id.</span></div> |
| <div class="line"><a name="l00269"></a><span class="lineno"> 269</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00270"></a><span class="lineno"> 270</span> <span class="comment"> */</span> </div> |
| <div class="line"><a name="l00271"></a><span class="lineno"> 271</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__get_routine_id</div> |
| <div class="line"><a name="l00272"></a><span class="lineno"> 272</span>  (</div> |
| <div class="line"><a name="l00273"></a><span class="lineno"> 273</span>  tree_table TEXT </div> |
| <div class="line"><a name="l00274"></a><span class="lineno"> 274</span>  ) </div> |
| <div class="line"><a name="l00275"></a><span class="lineno"> 275</span> RETURNS INT AS $$</div> |
| <div class="line"><a name="l00276"></a><span class="lineno"> 276</span> DECLARE</div> |
| <div class="line"><a name="l00277"></a><span class="lineno"> 277</span>  name TEXT;</div> |
| <div class="line"><a name="l00278"></a><span class="lineno"> 278</span> BEGIN</div> |
| <div class="line"><a name="l00279"></a><span class="lineno"> 279</span>  name = MADLIB_SCHEMA.__get_routine_name(tree_table);</div> |
| <div class="line"><a name="l00280"></a><span class="lineno"> 280</span>  </div> |
| <div class="line"><a name="l00281"></a><span class="lineno"> 281</span>  IF (name = 'ignore') THEN</div> |
| <div class="line"><a name="l00282"></a><span class="lineno"> 282</span>  RETURN 1;</div> |
| <div class="line"><a name="l00283"></a><span class="lineno"> 283</span>  ELSIF (name = 'explicit') THEN</div> |
| <div class="line"><a name="l00284"></a><span class="lineno"> 284</span>  RETURN 2;</div> |
| <div class="line"><a name="l00285"></a><span class="lineno"> 285</span>  ELSE</div> |
| <div class="line"><a name="l00286"></a><span class="lineno"> 286</span>  RAISE EXCEPTION '__get_routine_id: %', name; </div> |
| <div class="line"><a name="l00287"></a><span class="lineno"> 287</span>  END IF;</div> |
| <div class="line"><a name="l00288"></a><span class="lineno"> 288</span>  </div> |
| <div class="line"><a name="l00289"></a><span class="lineno"> 289</span> END</div> |
| <div class="line"><a name="l00290"></a><span class="lineno"> 290</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l00291"></a><span class="lineno"> 291</span> </div> |
| <div class="line"><a name="l00292"></a><span class="lineno"> 292</span> </div> |
| <div class="line"><a name="l00293"></a><span class="lineno"> 293</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00294"></a><span class="lineno"> 294</span> <span class="comment"> * @brief Get the unknown values processing routine name. </span></div> |
| <div class="line"><a name="l00295"></a><span class="lineno"> 295</span> <span class="comment"> * The valid routine name is 'ignore' or 'explicit'.</span></div> |
| <div class="line"><a name="l00296"></a><span class="lineno"> 296</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00297"></a><span class="lineno"> 297</span> <span class="comment"> * @param tree_table The full name of the tree table.</span></div> |
| <div class="line"><a name="l00298"></a><span class="lineno"> 298</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00299"></a><span class="lineno"> 299</span> <span class="comment"> * @return The encoded missing value processing routine name.</span></div> |
| <div class="line"><a name="l00300"></a><span class="lineno"> 300</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00301"></a><span class="lineno"> 301</span> <span class="comment"> */</span> </div> |
| <div class="line"><a name="l00302"></a><span class="lineno"> 302</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__get_routine_name</div> |
| <div class="line"><a name="l00303"></a><span class="lineno"> 303</span>  (</div> |
| <div class="line"><a name="l00304"></a><span class="lineno"> 304</span>  tree_table TEXT </div> |
| <div class="line"><a name="l00305"></a><span class="lineno"> 305</span>  ) </div> |
| <div class="line"><a name="l00306"></a><span class="lineno"> 306</span> RETURNS TEXT AS $$</div> |
| <div class="line"><a name="l00307"></a><span class="lineno"> 307</span> DECLARE</div> |
| <div class="line"><a name="l00308"></a><span class="lineno"> 308</span>  curstmt TEXT;</div> |
| <div class="line"><a name="l00309"></a><span class="lineno"> 309</span>  name TEXT;</div> |
| <div class="line"><a name="l00310"></a><span class="lineno"> 310</span> BEGIN</div> |
| <div class="line"><a name="l00311"></a><span class="lineno"> 311</span>  PERFORM MADLIB_SCHEMA.__assert_table</div> |
| <div class="line"><a name="l00312"></a><span class="lineno"> 312</span>  (</div> |
| <div class="line"><a name="l00313"></a><span class="lineno"> 313</span>  'MADLIB_SCHEMA.training_info',</div> |
| <div class="line"><a name="l00314"></a><span class="lineno"> 314</span>  't'</div> |
| <div class="line"><a name="l00315"></a><span class="lineno"> 315</span>  );</div> |
| <div class="line"><a name="l00316"></a><span class="lineno"> 316</span> </div> |
| <div class="line"><a name="l00317"></a><span class="lineno"> 317</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l00318"></a><span class="lineno"> 318</span>  (</div> |
| <div class="line"><a name="l00319"></a><span class="lineno"> 319</span>  'SELECT how2handle_missing_value </div> |
| <div class="line"><a name="l00320"></a><span class="lineno"> 320</span>  FROM MADLIB_SCHEMA.training_info </div> |
| <div class="line"><a name="l00321"></a><span class="lineno"> 321</span>  WHERE result_table_oid = ''%''::regclass',</div> |
| <div class="line"><a name="l00322"></a><span class="lineno"> 322</span>  tree_table</div> |
| <div class="line"><a name="l00323"></a><span class="lineno"> 323</span>  );</div> |
| <div class="line"><a name="l00324"></a><span class="lineno"> 324</span>  EXECUTE curstmt INTO name;</div> |
| <div class="line"><a name="l00325"></a><span class="lineno"> 325</span> </div> |
| <div class="line"><a name="l00326"></a><span class="lineno"> 326</span>  RETURN name;</div> |
| <div class="line"><a name="l00327"></a><span class="lineno"> 327</span> END</div> |
| <div class="line"><a name="l00328"></a><span class="lineno"> 328</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l00329"></a><span class="lineno"> 329</span> </div> |
| <div class="line"><a name="l00330"></a><span class="lineno"> 330</span> </div> |
| <div class="line"><a name="l00331"></a><span class="lineno"> 331</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00332"></a><span class="lineno"> 332</span> <span class="comment"> * @brief Get the name of the tree table from the encoded table name. </span></div> |
| <div class="line"><a name="l00333"></a><span class="lineno"> 333</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00334"></a><span class="lineno"> 334</span> <span class="comment"> * @param enc_table_name The encoded table name. </span></div> |
| <div class="line"><a name="l00335"></a><span class="lineno"> 335</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00336"></a><span class="lineno"> 336</span> <span class="comment"> * @return The full name of the tree table.</span></div> |
| <div class="line"><a name="l00337"></a><span class="lineno"> 337</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00338"></a><span class="lineno"> 338</span> <span class="comment"> */</span> </div> |
| <div class="line"><a name="l00339"></a><span class="lineno"> 339</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__get_tree_table_name</div> |
| <div class="line"><a name="l00340"></a><span class="lineno"> 340</span>  (</div> |
| <div class="line"><a name="l00341"></a><span class="lineno"> 341</span>  enc_table_name TEXT </div> |
| <div class="line"><a name="l00342"></a><span class="lineno"> 342</span>  ) </div> |
| <div class="line"><a name="l00343"></a><span class="lineno"> 343</span> RETURNS TEXT AS $$</div> |
| <div class="line"><a name="l00344"></a><span class="lineno"> 344</span> DECLARE</div> |
| <div class="line"><a name="l00345"></a><span class="lineno"> 345</span>  curstmt TEXT;</div> |
| <div class="line"><a name="l00346"></a><span class="lineno"> 346</span>  name TEXT;</div> |
| <div class="line"><a name="l00347"></a><span class="lineno"> 347</span> BEGIN</div> |
| <div class="line"><a name="l00348"></a><span class="lineno"> 348</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l00349"></a><span class="lineno"> 349</span>  (</div> |
| <div class="line"><a name="l00350"></a><span class="lineno"> 350</span>  'SELECT MADLIB_SCHEMA.__regclass_to_text(result_table_oid::regclass) </div> |
| <div class="line"><a name="l00351"></a><span class="lineno"> 351</span>  FROM MADLIB_SCHEMA.training_info</div> |
| <div class="line"><a name="l00352"></a><span class="lineno"> 352</span>  WHERE training_encoded_table_oid = ''%''::regclass</div> |
| <div class="line"><a name="l00353"></a><span class="lineno"> 353</span>  LIMIT 1',</div> |
| <div class="line"><a name="l00354"></a><span class="lineno"> 354</span>  enc_table_name</div> |
| <div class="line"><a name="l00355"></a><span class="lineno"> 355</span>  );</div> |
| <div class="line"><a name="l00356"></a><span class="lineno"> 356</span>  EXECUTE curstmt INTO name;</div> |
| <div class="line"><a name="l00357"></a><span class="lineno"> 357</span> </div> |
| <div class="line"><a name="l00358"></a><span class="lineno"> 358</span>  RETURN name;</div> |
| <div class="line"><a name="l00359"></a><span class="lineno"> 359</span> END</div> |
| <div class="line"><a name="l00360"></a><span class="lineno"> 360</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l00361"></a><span class="lineno"> 361</span> </div> |
| <div class="line"><a name="l00362"></a><span class="lineno"> 362</span> </div> |
| <div class="line"><a name="l00363"></a><span class="lineno"> 363</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__best_scv_sfunc</div> |
| <div class="line"><a name="l00364"></a><span class="lineno"> 364</span>  (</div> |
| <div class="line"><a name="l00365"></a><span class="lineno"> 365</span>  result FLOAT8[], -- intermediate result</div> |
| <div class="line"><a name="l00366"></a><span class="lineno"> 366</span>  scv FLOAT8[],</div> |
| <div class="line"><a name="l00367"></a><span class="lineno"> 367</span>  fid INT,</div> |
| <div class="line"><a name="l00368"></a><span class="lineno"> 368</span>  split_value FLOAT8</div> |
| <div class="line"><a name="l00369"></a><span class="lineno"> 369</span>  ) </div> |
| <div class="line"><a name="l00370"></a><span class="lineno"> 370</span> RETURNS FLOAT8[] </div> |
| <div class="line"><a name="l00371"></a><span class="lineno"> 371</span> AS 'MODULE_PATHNAME', 'dt_best_scv_sfunc'</div> |
| <div class="line"><a name="l00372"></a><span class="lineno"> 372</span> LANGUAGE C STRICT IMMUTABLE;</div> |
| <div class="line"><a name="l00373"></a><span class="lineno"> 373</span> </div> |
| <div class="line"><a name="l00374"></a><span class="lineno"> 374</span> </div> |
| <div class="line"><a name="l00375"></a><span class="lineno"> 375</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__best_scv_prefunc</div> |
| <div class="line"><a name="l00376"></a><span class="lineno"> 376</span>  (</div> |
| <div class="line"><a name="l00377"></a><span class="lineno"> 377</span>  sfunc1_result FLOAT8[],</div> |
| <div class="line"><a name="l00378"></a><span class="lineno"> 378</span>  sfunc2_result FLOAT8[]</div> |
| <div class="line"><a name="l00379"></a><span class="lineno"> 379</span>  ) </div> |
| <div class="line"><a name="l00380"></a><span class="lineno"> 380</span> RETURNS FLOAT8[]</div> |
| <div class="line"><a name="l00381"></a><span class="lineno"> 381</span> AS 'MODULE_PATHNAME', 'dt_best_scv_prefunc'</div> |
| <div class="line"><a name="l00382"></a><span class="lineno"> 382</span> LANGUAGE C STRICT IMMUTABLE;</div> |
| <div class="line"><a name="l00383"></a><span class="lineno"> 383</span> </div> |
| <div class="line"><a name="l00384"></a><span class="lineno"> 384</span> </div> |
| <div class="line"><a name="l00385"></a><span class="lineno"> 385</span> DROP AGGREGATE IF EXISTS MADLIB_SCHEMA.__best_scv_aggr</div> |
| <div class="line"><a name="l00386"></a><span class="lineno"> 386</span>  (</div> |
| <div class="line"><a name="l00387"></a><span class="lineno"> 387</span>  FLOAT8[], -- scv</div> |
| <div class="line"><a name="l00388"></a><span class="lineno"> 388</span>  INT, -- fid</div> |
| <div class="line"><a name="l00389"></a><span class="lineno"> 389</span>  FLOAT8 -- split_value </div> |
| <div class="line"><a name="l00390"></a><span class="lineno"> 390</span>  ) CASCADE;</div> |
| <div class="line"><a name="l00391"></a><span class="lineno"> 391</span> CREATE</div> |
| <div class="line"><a name="l00392"></a><span class="lineno"> 392</span> AGGREGATE MADLIB_SCHEMA.__best_scv_aggr</div> |
| <div class="line"><a name="l00393"></a><span class="lineno"> 393</span>  (</div> |
| <div class="line"><a name="l00394"></a><span class="lineno"> 394</span>  FLOAT8[], -- scv</div> |
| <div class="line"><a name="l00395"></a><span class="lineno"> 395</span>  INT, -- fid</div> |
| <div class="line"><a name="l00396"></a><span class="lineno"> 396</span>  FLOAT8 -- split_value</div> |
| <div class="line"><a name="l00397"></a><span class="lineno"> 397</span>  ) </div> |
| <div class="line"><a name="l00398"></a><span class="lineno"> 398</span> (</div> |
| <div class="line"><a name="l00399"></a><span class="lineno"> 399</span>  SFUNC=MADLIB_SCHEMA.__best_scv_sfunc,</div> |
| <div class="line"><a name="l00400"></a><span class="lineno"> 400</span>  m4_ifdef(`__GREENPLUM__', `prefunc=MADLIB_SCHEMA.__best_scv_prefunc,')</div> |
| <div class="line"><a name="l00401"></a><span class="lineno"> 401</span>  STYPE=FLOAT8[],</div> |
| <div class="line"><a name="l00402"></a><span class="lineno"> 402</span>  initcond = '{0, 0, 0, 0, 0, 0, 0}<span class="stringliteral">'</span></div> |
| <div class="line"><a name="l00403"></a><span class="lineno"> 403</span> <span class="stringliteral">);</span></div> |
| <div class="line"><a name="l00404"></a><span class="lineno"> 404</span> <span class="stringliteral"></span></div> |
| <div class="line"><a name="l00405"></a><span class="lineno"> 405</span> <span class="stringliteral"></span></div> |
| <div class="line"><a name="l00406"></a><span class="lineno"> 406</span> <span class="stringliteral">/*</span></div> |
| <div class="line"><a name="l00407"></a><span class="lineno"> 407</span> <span class="stringliteral"> * @brief The step function is defined to process each record in the ACS set. </span></div> |
| <div class="line"><a name="l00408"></a><span class="lineno"> 408</span> <span class="stringliteral"> * The records have this format: </span></div> |
| <div class="line"><a name="l00409"></a><span class="lineno"> 409</span> <span class="stringliteral"> * {fid, fval, is_cont, split_value, le, total, tid, nid}</span></div> |
| <div class="line"><a name="l00410"></a><span class="lineno"> 410</span> <span class="stringliteral"> *</span></div> |
| <div class="line"><a name="l00411"></a><span class="lineno"> 411</span> <span class="stringliteral"> * @param result The array used to keep the best attribute'</span>s info.</div> |
| <div class="line"><a name="l00412"></a><span class="lineno"> 412</span>  * @param sc_code The code of the split criterion.</div> |
| <div class="line"><a name="l00413"></a><span class="lineno"> 413</span>  * @param is_cont True - The feature is continuous. </div> |
| <div class="line"><a name="l00414"></a><span class="lineno"> 414</span>  * False - The feature is discrete.</div> |
| <div class="line"><a name="l00415"></a><span class="lineno"> 415</span>  * @param num_class The total number of classes.</div> |
| <div class="line"><a name="l00416"></a><span class="lineno"> 416</span>  * @param le_array The le component of the ACS record. le_array[i] is the </div> |
| <div class="line"><a name="l00417"></a><span class="lineno"> 417</span>  * number of samples whose <span class="keyword">class </span>code equals to i and </div> |
| <div class="line"><a name="l00418"></a><span class="lineno"> 418</span>  * whose fval is less-than or equal to the fval component </div> |
| <div class="line"><a name="l00419"></a><span class="lineno"> 419</span>  * of the ACS record being processed.</div> |
| <div class="line"><a name="l00420"></a><span class="lineno"> 420</span>  * @param total_array The total component of the ACS record. total_array[i] is</div> |
| <div class="line"><a name="l00421"></a><span class="lineno"> 421</span>  * the number of samples whose <span class="keyword">class </span>code equals to i.</div> |
| <div class="line"><a name="l00422"></a><span class="lineno"> 422</span>  * @param true_total The real total number of samples currently assigned to </div> |
| <div class="line"><a name="l00423"></a><span class="lineno"> 423</span>  * the node identified by (tid, nid). If there are missing</div> |
| <div class="line"><a name="l00424"></a><span class="lineno"> 424</span>  * values in fval, the sum of all elements in total_array</div> |
| <div class="line"><a name="l00425"></a><span class="lineno"> 425</span>  * will be less than true_total.</div> |
| <div class="line"><a name="l00426"></a><span class="lineno"> 426</span>  *</div> |
| <div class="line"><a name="l00427"></a><span class="lineno"> 427</span>  * @<span class="keywordflow">return</span> A 9-element array. Please refer to the definition of SCV_STATE_ARRAY_INDEX</div> |
| <div class="line"><a name="l00428"></a><span class="lineno"> 428</span>  * in dt.c <span class="keywordflow">for</span> the detailed information of <span class="keyword">this</span> array.</div> |
| <div class="line"><a name="l00429"></a><span class="lineno"> 429</span>  *</div> |
| <div class="line"><a name="l00430"></a><span class="lineno"> 430</span>  */</div> |
| <div class="line"><a name="l00431"></a><span class="lineno"> 431</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__scv_aggr_sfunc</div> |
| <div class="line"><a name="l00432"></a><span class="lineno"> 432</span>  (</div> |
| <div class="line"><a name="l00433"></a><span class="lineno"> 433</span>  result FLOAT8[],</div> |
| <div class="line"><a name="l00434"></a><span class="lineno"> 434</span>  sc_code INT,</div> |
| <div class="line"><a name="l00435"></a><span class="lineno"> 435</span>  is_cont BOOLEAN,</div> |
| <div class="line"><a name="l00436"></a><span class="lineno"> 436</span>  num_class INT,</div> |
| <div class="line"><a name="l00437"></a><span class="lineno"> 437</span>  le_array FLOAT8[],</div> |
| <div class="line"><a name="l00438"></a><span class="lineno"> 438</span>  total_array FLOAT8[],</div> |
| <div class="line"><a name="l00439"></a><span class="lineno"> 439</span>  true_total BIGINT</div> |
| <div class="line"><a name="l00440"></a><span class="lineno"> 440</span>  )</div> |
| <div class="line"><a name="l00441"></a><span class="lineno"> 441</span> RETURNS FLOAT8[]</div> |
| <div class="line"><a name="l00442"></a><span class="lineno"> 442</span> AS <span class="stringliteral">'MODULE_PATHNAME'</span>, <span class="stringliteral">'dt_scv_aggr_sfunc'</span></div> |
| <div class="line"><a name="l00443"></a><span class="lineno"> 443</span> LANGUAGE C IMMUTABLE;</div> |
| <div class="line"><a name="l00444"></a><span class="lineno"> 444</span> </div> |
| <div class="line"><a name="l00445"></a><span class="lineno"> 445</span> </div> |
| <div class="line"><a name="l00446"></a><span class="lineno"> 446</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00447"></a><span class="lineno"> 447</span> <span class="comment"> * @brief The pre-function for the aggregation of splitting criteria values. It </span></div> |
| <div class="line"><a name="l00448"></a><span class="lineno"> 448</span> <span class="comment"> * takes the state array produced by two sfunc and combine them together.</span></div> |
| <div class="line"><a name="l00449"></a><span class="lineno"> 449</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00450"></a><span class="lineno"> 450</span> <span class="comment"> * @param sfunc1_result The array from sfunc1.</span></div> |
| <div class="line"><a name="l00451"></a><span class="lineno"> 451</span> <span class="comment"> * @param sfunc2_result The array from sfunc2.</span></div> |
| <div class="line"><a name="l00452"></a><span class="lineno"> 452</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00453"></a><span class="lineno"> 453</span> <span class="comment"> * @return A 9-element array. Please refer to the definition of SCV_STATE_ARRAY_INDEX</span></div> |
| <div class="line"><a name="l00454"></a><span class="lineno"> 454</span> <span class="comment"> * in dt.c for the detailed information of this array.</span></div> |
| <div class="line"><a name="l00455"></a><span class="lineno"> 455</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00456"></a><span class="lineno"> 456</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00457"></a><span class="lineno"> 457</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__scv_aggr_prefunc</div> |
| <div class="line"><a name="l00458"></a><span class="lineno"> 458</span>  (</div> |
| <div class="line"><a name="l00459"></a><span class="lineno"> 459</span>  sfunc1_result FLOAT8[],</div> |
| <div class="line"><a name="l00460"></a><span class="lineno"> 460</span>  sfunc2_result FLOAT8[]</div> |
| <div class="line"><a name="l00461"></a><span class="lineno"> 461</span>  ) </div> |
| <div class="line"><a name="l00462"></a><span class="lineno"> 462</span> RETURNS FLOAT8[]</div> |
| <div class="line"><a name="l00463"></a><span class="lineno"> 463</span> AS <span class="stringliteral">'MODULE_PATHNAME'</span>, <span class="stringliteral">'dt_scv_aggr_prefunc'</span></div> |
| <div class="line"><a name="l00464"></a><span class="lineno"> 464</span> LANGUAGE C STRICT IMMUTABLE;</div> |
| <div class="line"><a name="l00465"></a><span class="lineno"> 465</span> </div> |
| <div class="line"><a name="l00466"></a><span class="lineno"> 466</span> </div> |
| <div class="line"><a name="l00467"></a><span class="lineno"> 467</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00468"></a><span class="lineno"> 468</span> <span class="comment"> * @brief The final function for the aggregation of splitting criteria values.</span></div> |
| <div class="line"><a name="l00469"></a><span class="lineno"> 469</span> <span class="comment"> * It takes the state array produced by the sfunc and produces a</span></div> |
| <div class="line"><a name="l00470"></a><span class="lineno"> 470</span> <span class="comment"> * 5-element array.</span></div> |
| <div class="line"><a name="l00471"></a><span class="lineno"> 471</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00472"></a><span class="lineno"> 472</span> <span class="comment"> * @param internal_result The 9-element array produced by dt_scv_aggr_prefunc</span></div> |
| <div class="line"><a name="l00473"></a><span class="lineno"> 473</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00474"></a><span class="lineno"> 474</span> <span class="comment"> * @return A 5-element array. Please refer to the definition of SCV_FINAL_ARRAY_INDEX</span></div> |
| <div class="line"><a name="l00475"></a><span class="lineno"> 475</span> <span class="comment"> * in dt.c for the detailed information of this array.</span></div> |
| <div class="line"><a name="l00476"></a><span class="lineno"> 476</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00477"></a><span class="lineno"> 477</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00478"></a><span class="lineno"> 478</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__scv_aggr_ffunc</div> |
| <div class="line"><a name="l00479"></a><span class="lineno"> 479</span>  (</div> |
| <div class="line"><a name="l00480"></a><span class="lineno"> 480</span>  internal_result FLOAT8[]</div> |
| <div class="line"><a name="l00481"></a><span class="lineno"> 481</span>  ) </div> |
| <div class="line"><a name="l00482"></a><span class="lineno"> 482</span> RETURNS FLOAT8[]</div> |
| <div class="line"><a name="l00483"></a><span class="lineno"> 483</span> AS <span class="stringliteral">'MODULE_PATHNAME'</span>, <span class="stringliteral">'dt_scv_aggr_ffunc'</span></div> |
| <div class="line"><a name="l00484"></a><span class="lineno"> 484</span> LANGUAGE C STRICT IMMUTABLE;</div> |
| <div class="line"><a name="l00485"></a><span class="lineno"> 485</span> </div> |
| <div class="line"><a name="l00486"></a><span class="lineno"> 486</span> </div> |
| <div class="line"><a name="l00487"></a><span class="lineno"> 487</span> DROP AGGREGATE IF EXISTS MADLIB_SCHEMA.__scv_aggr</div> |
| <div class="line"><a name="l00488"></a><span class="lineno"> 488</span>  (</div> |
| <div class="line"><a name="l00489"></a><span class="lineno"> 489</span>  INT, -- sc</div> |
| <div class="line"><a name="l00490"></a><span class="lineno"> 490</span>  BOOLEAN, -- is_cont</div> |
| <div class="line"><a name="l00491"></a><span class="lineno"> 491</span>  INT, -- total number of classes</div> |
| <div class="line"><a name="l00492"></a><span class="lineno"> 492</span>  FLOAT8[], -- le array</div> |
| <div class="line"><a name="l00493"></a><span class="lineno"> 493</span>  FLOAT8[], -- total count array</div> |
| <div class="line"><a name="l00494"></a><span class="lineno"> 494</span>  BIGINT -- the total number of samples</div> |
| <div class="line"><a name="l00495"></a><span class="lineno"> 495</span>  ) CASCADE;</div> |
| <div class="line"><a name="l00496"></a><span class="lineno"> 496</span> CREATE</div> |
| <div class="line"><a name="l00497"></a><span class="lineno"> 497</span> AGGREGATE MADLIB_SCHEMA.__scv_aggr</div> |
| <div class="line"><a name="l00498"></a><span class="lineno"> 498</span>  (</div> |
| <div class="line"><a name="l00499"></a><span class="lineno"> 499</span>  INT, -- sc</div> |
| <div class="line"><a name="l00500"></a><span class="lineno"> 500</span>  BOOLEAN, -- is_cont</div> |
| <div class="line"><a name="l00501"></a><span class="lineno"> 501</span>  INT, -- total number of classes</div> |
| <div class="line"><a name="l00502"></a><span class="lineno"> 502</span>  FLOAT8[], -- le array</div> |
| <div class="line"><a name="l00503"></a><span class="lineno"> 503</span>  FLOAT8[], -- total count array</div> |
| <div class="line"><a name="l00504"></a><span class="lineno"> 504</span>  BIGINT -- the total number of samples</div> |
| <div class="line"><a name="l00505"></a><span class="lineno"> 505</span>  ) </div> |
| <div class="line"><a name="l00506"></a><span class="lineno"> 506</span> (</div> |
| <div class="line"><a name="l00507"></a><span class="lineno"> 507</span>  SFUNC=MADLIB_SCHEMA.__scv_aggr_sfunc,</div> |
| <div class="line"><a name="l00508"></a><span class="lineno"> 508</span>  m4_ifdef(`__GREENPLUM__<span class="stringliteral">', `prefunc=MADLIB_SCHEMA.__scv_aggr_prefunc,'</span>)</div> |
| <div class="line"><a name="l00509"></a><span class="lineno"> 509</span>  FINALFUNC=MADLIB_SCHEMA.__scv_aggr_ffunc,</div> |
| <div class="line"><a name="l00510"></a><span class="lineno"> 510</span>  STYPE=FLOAT8[],</div> |
| <div class="line"><a name="l00511"></a><span class="lineno"> 511</span>  initcond = '{0, 0, 0, 0, 0, 0, 0, 0, 0}<span class="stringliteral">'</span></div> |
| <div class="line"><a name="l00512"></a><span class="lineno"> 512</span> <span class="stringliteral"> -- 1 sc: 1 infogain, 2 gainratio, 3 gini</span></div> |
| <div class="line"><a name="l00513"></a><span class="lineno"> 513</span> <span class="stringliteral"> -- 2 is_cont</span></div> |
| <div class="line"><a name="l00514"></a><span class="lineno"> 514</span> <span class="stringliteral"> -- 3 scv_class_info</span></div> |
| <div class="line"><a name="l00515"></a><span class="lineno"> 515</span> <span class="stringliteral"> -- 4 scv_attr_info</span></div> |
| <div class="line"><a name="l00516"></a><span class="lineno"> 516</span> <span class="stringliteral"> -- 5 scv_class_attr_info</span></div> |
| <div class="line"><a name="l00517"></a><span class="lineno"> 517</span> <span class="stringliteral"> -- 6 scv_count</span></div> |
| <div class="line"><a name="l00518"></a><span class="lineno"> 518</span> <span class="stringliteral"> -- 7 scv_total</span></div> |
| <div class="line"><a name="l00519"></a><span class="lineno"> 519</span> <span class="stringliteral"> -- 8 max_class_id</span></div> |
| <div class="line"><a name="l00520"></a><span class="lineno"> 520</span> <span class="stringliteral"> -- 9 max_class_count</span></div> |
| <div class="line"><a name="l00521"></a><span class="lineno"> 521</span> <span class="stringliteral">);</span></div> |
| <div class="line"><a name="l00522"></a><span class="lineno"> 522</span> <span class="stringliteral"></span></div> |
| <div class="line"><a name="l00523"></a><span class="lineno"> 523</span> <span class="stringliteral"></span></div> |
| <div class="line"><a name="l00524"></a><span class="lineno"> 524</span> <span class="stringliteral">/*</span></div> |
| <div class="line"><a name="l00525"></a><span class="lineno"> 525</span> <span class="stringliteral"> * @brief Retrieve the specified number of unique features for a node.</span></div> |
| <div class="line"><a name="l00526"></a><span class="lineno"> 526</span> <span class="stringliteral"> * Discrete features used by ancestor nodes will be excluded.</span></div> |
| <div class="line"><a name="l00527"></a><span class="lineno"> 527</span> <span class="stringliteral"> * If the number of remaining features is less or equal than the</span></div> |
| <div class="line"><a name="l00528"></a><span class="lineno"> 528</span> <span class="stringliteral"> * requested number of features, then all the remaining features</span></div> |
| <div class="line"><a name="l00529"></a><span class="lineno"> 529</span> <span class="stringliteral"> * will be returned. Otherwise, we will sample the requested </span></div> |
| <div class="line"><a name="l00530"></a><span class="lineno"> 530</span> <span class="stringliteral"> * number of features from the remaining features.</span></div> |
| <div class="line"><a name="l00531"></a><span class="lineno"> 531</span> <span class="stringliteral"> *</span></div> |
| <div class="line"><a name="l00532"></a><span class="lineno"> 532</span> <span class="stringliteral"> * @param num_req_features The number of requested features.</span></div> |
| <div class="line"><a name="l00533"></a><span class="lineno"> 533</span> <span class="stringliteral"> * @param num_features The total number of features.</span></div> |
| <div class="line"><a name="l00534"></a><span class="lineno"> 534</span> <span class="stringliteral"> * @param nid The ID of the node for which the</span></div> |
| <div class="line"><a name="l00535"></a><span class="lineno"> 535</span> <span class="stringliteral"> * features are sampled.</span></div> |
| <div class="line"><a name="l00536"></a><span class="lineno"> 536</span> <span class="stringliteral"> * @param dp_fids The IDs of the discrete features</span></div> |
| <div class="line"><a name="l00537"></a><span class="lineno"> 537</span> <span class="stringliteral"> * used by the ancestors.</span></div> |
| <div class="line"><a name="l00538"></a><span class="lineno"> 538</span> <span class="stringliteral"> *</span></div> |
| <div class="line"><a name="l00539"></a><span class="lineno"> 539</span> <span class="stringliteral"> * @return An array containing all the IDs of chosen features.</span></div> |
| <div class="line"><a name="l00540"></a><span class="lineno"> 540</span> <span class="stringliteral"> *</span></div> |
| <div class="line"><a name="l00541"></a><span class="lineno"> 541</span> <span class="stringliteral"> */</span></div> |
| <div class="line"><a name="l00542"></a><span class="lineno"> 542</span> <span class="stringliteral">CREATE OR REPLACE FUNCTION </span></div> |
| <div class="line"><a name="l00543"></a><span class="lineno"> 543</span> <span class="stringliteral">MADLIB_SCHEMA.__dt_get_node_split_fids(INT4, INT4, INT4, INT4[])</span></div> |
| <div class="line"><a name="l00544"></a><span class="lineno"> 544</span> <span class="stringliteral">RETURNS INT[]</span></div> |
| <div class="line"><a name="l00545"></a><span class="lineno"> 545</span> <span class="stringliteral">AS '</span>MODULE_PATHNAME<span class="stringliteral">', '</span>dt_get_node_split_fids<span class="stringliteral">'</span></div> |
| <div class="line"><a name="l00546"></a><span class="lineno"> 546</span> <span class="stringliteral">LANGUAGE C VOLATILE;</span></div> |
| <div class="line"><a name="l00547"></a><span class="lineno"> 547</span> <span class="stringliteral"></span></div> |
| <div class="line"><a name="l00548"></a><span class="lineno"> 548</span> <span class="stringliteral"></span></div> |
| <div class="line"><a name="l00549"></a><span class="lineno"> 549</span> <span class="stringliteral">/*</span></div> |
| <div class="line"><a name="l00550"></a><span class="lineno"> 550</span> <span class="stringliteral"> * @brief Retrieve the selected features for a node. We will create a table, named </span></div> |
| <div class="line"><a name="l00551"></a><span class="lineno"> 551</span> <span class="stringliteral"> * sf_association, to store the association between selected feature IDs and</span></div> |
| <div class="line"><a name="l00552"></a><span class="lineno"> 552</span> <span class="stringliteral"> * node IDs.</span></div> |
| <div class="line"><a name="l00553"></a><span class="lineno"> 553</span> <span class="stringliteral"> *</span></div> |
| <div class="line"><a name="l00554"></a><span class="lineno"> 554</span> <span class="stringliteral"> * @param nid_table_name The full name of the table which contains all the </span></div> |
| <div class="line"><a name="l00555"></a><span class="lineno"> 555</span> <span class="stringliteral"> * node IDs.</span></div> |
| <div class="line"><a name="l00556"></a><span class="lineno"> 556</span> <span class="stringliteral"> * @param result_table_name The full name of the table which contains the parent</span></div> |
| <div class="line"><a name="l00557"></a><span class="lineno"> 557</span> <span class="stringliteral"> * discrete features for each node.</span></div> |
| <div class="line"><a name="l00558"></a><span class="lineno"> 558</span> <span class="stringliteral"> * @param num_chosen_fids The number of feature IDs will be chosen for a node.</span></div> |
| <div class="line"><a name="l00559"></a><span class="lineno"> 559</span> <span class="stringliteral"> * @param total_num_fids The total number of feature IDs, total_num_fids </span></div> |
| <div class="line"><a name="l00560"></a><span class="lineno"> 560</span> <span class="stringliteral"> * >= num_chosen_fids.</span></div> |
| <div class="line"><a name="l00561"></a><span class="lineno"> 561</span> <span class="stringliteral"> * If num_chosen_fids < total_num_fids, then we will </span></div> |
| <div class="line"><a name="l00562"></a><span class="lineno"> 562</span> <span class="stringliteral"> * randomly select num_chosen_fids features from all</span></div> |
| <div class="line"><a name="l00563"></a><span class="lineno"> 563</span> <span class="stringliteral"> * the features. Otherwise, we will return all the </span></div> |
| <div class="line"><a name="l00564"></a><span class="lineno"> 564</span> <span class="stringliteral"> * features exception they belong to the parent discrete</span></div> |
| <div class="line"><a name="l00565"></a><span class="lineno"> 565</span> <span class="stringliteral"> * features for a node.</span></div> |
| <div class="line"><a name="l00566"></a><span class="lineno"> 566</span> <span class="stringliteral"> * @param verbosity > 0 means this function runs in verbose mode.</span></div> |
| <div class="line"><a name="l00567"></a><span class="lineno"> 567</span> <span class="stringliteral"> * </span></div> |
| <div class="line"><a name="l00568"></a><span class="lineno"> 568</span> <span class="stringliteral"> * @return An constant string for the association table name.</span></div> |
| <div class="line"><a name="l00569"></a><span class="lineno"> 569</span> <span class="stringliteral"> *</span></div> |
| <div class="line"><a name="l00570"></a><span class="lineno"> 570</span> <span class="stringliteral"> */</span></div> |
| <div class="line"><a name="l00571"></a><span class="lineno"> 571</span> <span class="stringliteral">CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__get_features_of_nodes</span></div> |
| <div class="line"><a name="l00572"></a><span class="lineno"> 572</span> <span class="stringliteral"> (</span></div> |
| <div class="line"><a name="l00573"></a><span class="lineno"> 573</span> <span class="stringliteral"> nid_table_name TEXT,</span></div> |
| <div class="line"><a name="l00574"></a><span class="lineno"> 574</span> <span class="stringliteral"> result_table_name TEXT,</span></div> |
| <div class="line"><a name="l00575"></a><span class="lineno"> 575</span> <span class="stringliteral"> num_chosen_fids INT,</span></div> |
| <div class="line"><a name="l00576"></a><span class="lineno"> 576</span> <span class="stringliteral"> total_num_fids INT,</span></div> |
| <div class="line"><a name="l00577"></a><span class="lineno"> 577</span> <span class="stringliteral"> verbosity INT</span></div> |
| <div class="line"><a name="l00578"></a><span class="lineno"> 578</span> <span class="stringliteral"> )</span></div> |
| <div class="line"><a name="l00579"></a><span class="lineno"> 579</span> <span class="stringliteral">RETURNS TEXT AS $$</span></div> |
| <div class="line"><a name="l00580"></a><span class="lineno"> 580</span> <span class="stringliteral">DECLARE</span></div> |
| <div class="line"><a name="l00581"></a><span class="lineno"> 581</span> <span class="stringliteral"> curstmt TEXT;</span></div> |
| <div class="line"><a name="l00582"></a><span class="lineno"> 582</span> <span class="stringliteral">BEGIN</span></div> |
| <div class="line"><a name="l00583"></a><span class="lineno"> 583</span> <span class="stringliteral"> -- The sf_association table records which features are used</span></div> |
| <div class="line"><a name="l00584"></a><span class="lineno"> 584</span> <span class="stringliteral"> -- for finding the best split for a node.</span></div> |
| <div class="line"><a name="l00585"></a><span class="lineno"> 585</span> <span class="stringliteral"> -- It has two columns:</span></div> |
| <div class="line"><a name="l00586"></a><span class="lineno"> 586</span> <span class="stringliteral"> -- nid -- The id of a node.</span></div> |
| <div class="line"><a name="l00587"></a><span class="lineno"> 587</span> <span class="stringliteral"> -- fid -- The id of a feature.</span></div> |
| <div class="line"><a name="l00588"></a><span class="lineno"> 588</span> <span class="stringliteral"> EXECUTE '</span>TRUNCATE sf_assoc<span class="stringliteral">';</span></div> |
| <div class="line"><a name="l00589"></a><span class="lineno"> 589</span> <span class="stringliteral"> </span></div> |
| <div class="line"><a name="l00590"></a><span class="lineno"> 590</span> <span class="stringliteral"> curstmt = MADLIB_SCHEMA.__format</span></div> |
| <div class="line"><a name="l00591"></a><span class="lineno"> 591</span> <span class="stringliteral"> (</span></div> |
| <div class="line"><a name="l00592"></a><span class="lineno"> 592</span> <span class="stringliteral"> '</span>INSERT INTO sf_assoc(nid, fid) </div> |
| <div class="line"><a name="l00593"></a><span class="lineno"> 593</span>  SELECT </div> |
| <div class="line"><a name="l00594"></a><span class="lineno"> 594</span>  nid, </div> |
| <div class="line"><a name="l00595"></a><span class="lineno"> 595</span>  unnest(MADLIB_SCHEMA.__dt_get_node_split_fids(%, %,</div> |
| <div class="line"><a name="l00596"></a><span class="lineno"> 596</span>  nid,dp_ids)) as fid</div> |
| <div class="line"><a name="l00597"></a><span class="lineno"> 597</span>  FROM (SELECT nid, dp_ids </div> |
| <div class="line"><a name="l00598"></a><span class="lineno"> 598</span>  FROM % s1, % s2 </div> |
| <div class="line"><a name="l00599"></a><span class="lineno"> 599</span>  WHERE s1.nid = s2.<span class="keywordtype">id</span></div> |
| <div class="line"><a name="l00600"></a><span class="lineno"> 600</span>  GROUP BY nid, dp_ids) t',</div> |
| <div class="line"><a name="l00601"></a><span class="lineno"> 601</span>  ARRAY[</div> |
| <div class="line"><a name="l00602"></a><span class="lineno"> 602</span>  num_chosen_fids::TEXT,</div> |
| <div class="line"><a name="l00603"></a><span class="lineno"> 603</span>  total_num_fids::TEXT,</div> |
| <div class="line"><a name="l00604"></a><span class="lineno"> 604</span>  nid_table_name,</div> |
| <div class="line"><a name="l00605"></a><span class="lineno"> 605</span>  result_table_name</div> |
| <div class="line"><a name="l00606"></a><span class="lineno"> 606</span>  ]</div> |
| <div class="line"><a name="l00607"></a><span class="lineno"> 607</span>  );</div> |
| <div class="line"><a name="l00608"></a><span class="lineno"> 608</span>  </div> |
| <div class="line"><a name="l00609"></a><span class="lineno"> 609</span>  IF (verbosity > 0) THEN</div> |
| <div class="line"><a name="l00610"></a><span class="lineno"> 610</span>  RAISE INFO 'build sample feature association stmt: %', curstmt;</div> |
| <div class="line"><a name="l00611"></a><span class="lineno"> 611</span>  END IF;</div> |
| <div class="line"><a name="l00612"></a><span class="lineno"> 612</span>  </div> |
| <div class="line"><a name="l00613"></a><span class="lineno"> 613</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l00614"></a><span class="lineno"> 614</span>  </div> |
| <div class="line"><a name="l00615"></a><span class="lineno"> 615</span>  -- we return an constant <span class="keywordtype">string</span> for the association table name</div> |
| <div class="line"><a name="l00616"></a><span class="lineno"> 616</span>  return 'sf_assoc';</div> |
| <div class="line"><a name="l00617"></a><span class="lineno"> 617</span>  </div> |
| <div class="line"><a name="l00618"></a><span class="lineno"> 618</span> END</div> |
| <div class="line"><a name="l00619"></a><span class="lineno"> 619</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l00620"></a><span class="lineno"> 620</span> </div> |
| <div class="line"><a name="l00621"></a><span class="lineno"> 621</span> </div> |
| <div class="line"><a name="l00622"></a><span class="lineno"> 622</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00623"></a><span class="lineno"> 623</span> <span class="comment"> * This UDT is used to keep the times of generating acc.</span></div> |
| <div class="line"><a name="l00624"></a><span class="lineno"> 624</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00625"></a><span class="lineno"> 625</span> <span class="comment"> * calc_pre_time The time of pre-processing.</span></div> |
| <div class="line"><a name="l00626"></a><span class="lineno"> 626</span> <span class="comment"> * calc_acc_time The time of calculating acc.</span></div> |
| <div class="line"><a name="l00627"></a><span class="lineno"> 627</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00628"></a><span class="lineno"> 628</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00629"></a><span class="lineno"> 629</span> DROP TYPE IF EXISTS MADLIB_SCHEMA.__gen_acc_time;</div> |
| <div class="line"><a name="l00630"></a><span class="lineno"> 630</span> CREATE TYPE MADLIB_SCHEMA.__gen_acc_time AS </div> |
| <div class="line"><a name="l00631"></a><span class="lineno"> 631</span> ( </div> |
| <div class="line"><a name="l00632"></a><span class="lineno"> 632</span>  calc_pre_time INTERVAL,</div> |
| <div class="line"><a name="l00633"></a><span class="lineno"> 633</span>  calc_acc_time INTERVAL</div> |
| <div class="line"><a name="l00634"></a><span class="lineno"> 634</span> );</div> |
| <div class="line"><a name="l00635"></a><span class="lineno"> 635</span> </div> |
| <div class="line"><a name="l00636"></a><span class="lineno"> 636</span> </div> |
| <div class="line"><a name="l00637"></a><span class="lineno"> 637</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00638"></a><span class="lineno"> 638</span> <span class="comment"> * @brief Generate the ACC for current leaf nodes.</span></div> |
| <div class="line"><a name="l00639"></a><span class="lineno"> 639</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00640"></a><span class="lineno"> 640</span> <span class="comment"> * @param encoded_table_name The full name of the encoded table for the </span></div> |
| <div class="line"><a name="l00641"></a><span class="lineno"> 641</span> <span class="comment"> * training table.</span></div> |
| <div class="line"><a name="l00642"></a><span class="lineno"> 642</span> <span class="comment"> * @param metatable_name The full name of the metatable contains the </span></div> |
| <div class="line"><a name="l00643"></a><span class="lineno"> 643</span> <span class="comment"> * relevant information of the input table.</span></div> |
| <div class="line"><a name="l00644"></a><span class="lineno"> 644</span> <span class="comment"> * @param result_table_name The full name of the training result table.</span></div> |
| <div class="line"><a name="l00645"></a><span class="lineno"> 645</span> <span class="comment"> * @param num_featrue_try The number of features will be chosen per node. </span></div> |
| <div class="line"><a name="l00646"></a><span class="lineno"> 646</span> <span class="comment"> * @param num_classes Total number of classes in training set.</span></div> |
| <div class="line"><a name="l00647"></a><span class="lineno"> 647</span> <span class="comment"> * @param verbosity > 0 means this function runs in verbose mode. </span></div> |
| <div class="line"><a name="l00648"></a><span class="lineno"> 648</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00649"></a><span class="lineno"> 649</span> <span class="comment"> * @return The time information for generating ACC.</span></div> |
| <div class="line"><a name="l00650"></a><span class="lineno"> 650</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00651"></a><span class="lineno"> 651</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00652"></a><span class="lineno"> 652</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__gen_acc</div> |
| <div class="line"><a name="l00653"></a><span class="lineno"> 653</span>  (</div> |
| <div class="line"><a name="l00654"></a><span class="lineno"> 654</span>  encoded_table_name TEXT,</div> |
| <div class="line"><a name="l00655"></a><span class="lineno"> 655</span>  metatable_name TEXT,</div> |
| <div class="line"><a name="l00656"></a><span class="lineno"> 656</span>  result_table_name TEXT,</div> |
| <div class="line"><a name="l00657"></a><span class="lineno"> 657</span>  tr_table_name TEXT,</div> |
| <div class="line"><a name="l00658"></a><span class="lineno"> 658</span>  sf_table_name TEXT,</div> |
| <div class="line"><a name="l00659"></a><span class="lineno"> 659</span>  num_featrue_try INT,</div> |
| <div class="line"><a name="l00660"></a><span class="lineno"> 660</span>  num_classes INT,</div> |
| <div class="line"><a name="l00661"></a><span class="lineno"> 661</span>  sampling_needed BOOLEAN,</div> |
| <div class="line"><a name="l00662"></a><span class="lineno"> 662</span>  verbosity INT</div> |
| <div class="line"><a name="l00663"></a><span class="lineno"> 663</span>  )</div> |
| <div class="line"><a name="l00664"></a><span class="lineno"> 664</span> RETURNS MADLIB_SCHEMA.__gen_acc_time AS $$</div> |
| <div class="line"><a name="l00665"></a><span class="lineno"> 665</span> DECLARE</div> |
| <div class="line"><a name="l00666"></a><span class="lineno"> 666</span>  curstmt TEXT := '';</div> |
| <div class="line"><a name="l00667"></a><span class="lineno"> 667</span>  num_fids INT := 1;</div> |
| <div class="line"><a name="l00668"></a><span class="lineno"> 668</span>  begin_calc_acc TIMESTAMP;</div> |
| <div class="line"><a name="l00669"></a><span class="lineno"> 669</span>  begin_calc_pre TIMESTAMP;</div> |
| <div class="line"><a name="l00670"></a><span class="lineno"> 670</span>  ret MADLIB_SCHEMA.__gen_acc_time;</div> |
| <div class="line"><a name="l00671"></a><span class="lineno"> 671</span>  select_stmt TEXT;</div> |
| <div class="line"><a name="l00672"></a><span class="lineno"> 672</span> BEGIN</div> |
| <div class="line"><a name="l00673"></a><span class="lineno"> 673</span>  begin_calc_pre = clock_timestamp();</div> |
| <div class="line"><a name="l00674"></a><span class="lineno"> 674</span> </div> |
| <div class="line"><a name="l00675"></a><span class="lineno"> 675</span>  -- get the number of features</div> |
| <div class="line"><a name="l00676"></a><span class="lineno"> 676</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l00677"></a><span class="lineno"> 677</span>  (</div> |
| <div class="line"><a name="l00678"></a><span class="lineno"> 678</span>  'SELECT COUNT(<span class="keywordtype">id</span>) </div> |
| <div class="line"><a name="l00679"></a><span class="lineno"> 679</span>  FROM % </div> |
| <div class="line"><a name="l00680"></a><span class="lineno"> 680</span>  WHERE column_type = ''f''',</div> |
| <div class="line"><a name="l00681"></a><span class="lineno"> 681</span>  metatable_name</div> |
| <div class="line"><a name="l00682"></a><span class="lineno"> 682</span>  );</div> |
| <div class="line"><a name="l00683"></a><span class="lineno"> 683</span>  EXECUTE curstmt INTO num_fids;</div> |
| <div class="line"><a name="l00684"></a><span class="lineno"> 684</span> </div> |
| <div class="line"><a name="l00685"></a><span class="lineno"> 685</span>  -- preprocessing time</div> |
| <div class="line"><a name="l00686"></a><span class="lineno"> 686</span>  ret.calc_pre_time = clock_timestamp() - begin_calc_pre;</div> |
| <div class="line"><a name="l00687"></a><span class="lineno"> 687</span>  begin_calc_acc = clock_timestamp();</div> |
| <div class="line"><a name="l00688"></a><span class="lineno"> 688</span>  </div> |
| <div class="line"><a name="l00689"></a><span class="lineno"> 689</span>  IF (sampling_needed) THEN</div> |
| <div class="line"><a name="l00690"></a><span class="lineno"> 690</span>  PERFORM MADLIB_SCHEMA.__get_features_of_nodes</div> |
| <div class="line"><a name="l00691"></a><span class="lineno"> 691</span>  (</div> |
| <div class="line"><a name="l00692"></a><span class="lineno"> 692</span>  tr_table_name,</div> |
| <div class="line"><a name="l00693"></a><span class="lineno"> 693</span>  result_table_name,</div> |
| <div class="line"><a name="l00694"></a><span class="lineno"> 694</span>  num_featrue_try,</div> |
| <div class="line"><a name="l00695"></a><span class="lineno"> 695</span>  num_fids,</div> |
| <div class="line"><a name="l00696"></a><span class="lineno"> 696</span>  verbosity</div> |
| <div class="line"><a name="l00697"></a><span class="lineno"> 697</span>  );</div> |
| <div class="line"><a name="l00698"></a><span class="lineno"> 698</span>  </div> |
| <div class="line"><a name="l00699"></a><span class="lineno"> 699</span>  select_stmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l00700"></a><span class="lineno"> 700</span>  (</div> |
| <div class="line"><a name="l00701"></a><span class="lineno"> 701</span>  'SELECT tr.tid, tr.nid, ed.fid, ed.fval, ed.is_cont, </div> |
| <div class="line"><a name="l00702"></a><span class="lineno"> 702</span>  ed.class, sum(weight) as count</div> |
| <div class="line"><a name="l00703"></a><span class="lineno"> 703</span>  FROM % ed, % tr, % sf</div> |
| <div class="line"><a name="l00704"></a><span class="lineno"> 704</span>  WHERE tr.nid = sf.nid AND ed.fid = sf.fid AND ed.<span class="keywordtype">id</span> = tr.<span class="keywordtype">id</span></div> |
| <div class="line"><a name="l00705"></a><span class="lineno"> 705</span>  GROUP BY tr.tid, tr.nid, ed.fid, ed.fval, </div> |
| <div class="line"><a name="l00706"></a><span class="lineno"> 706</span>  ed.is_cont, ed.class',</div> |
| <div class="line"><a name="l00707"></a><span class="lineno"> 707</span>  ARRAY[</div> |
| <div class="line"><a name="l00708"></a><span class="lineno"> 708</span>  encoded_table_name,</div> |
| <div class="line"><a name="l00709"></a><span class="lineno"> 709</span>  tr_table_name,</div> |
| <div class="line"><a name="l00710"></a><span class="lineno"> 710</span>  sf_table_name</div> |
| <div class="line"><a name="l00711"></a><span class="lineno"> 711</span>  ]</div> |
| <div class="line"><a name="l00712"></a><span class="lineno"> 712</span>  );</div> |
| <div class="line"><a name="l00713"></a><span class="lineno"> 713</span>  ELSE</div> |
| <div class="line"><a name="l00714"></a><span class="lineno"> 714</span>  select_stmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l00715"></a><span class="lineno"> 715</span>  (</div> |
| <div class="line"><a name="l00716"></a><span class="lineno"> 716</span>  'SELECT tr.tid, tr.nid, ed.fid, ed.fval, ed.is_cont, </div> |
| <div class="line"><a name="l00717"></a><span class="lineno"> 717</span>  ed.class, sum(weight) as count</div> |
| <div class="line"><a name="l00718"></a><span class="lineno"> 718</span>  FROM % ed, % tr</div> |
| <div class="line"><a name="l00719"></a><span class="lineno"> 719</span>  WHERE ed.<span class="keywordtype">id</span> = tr.<span class="keywordtype">id</span></div> |
| <div class="line"><a name="l00720"></a><span class="lineno"> 720</span>  GROUP BY tr.tid, tr.nid, ed.fid, ed.fval, </div> |
| <div class="line"><a name="l00721"></a><span class="lineno"> 721</span>  ed.is_cont, ed.class',</div> |
| <div class="line"><a name="l00722"></a><span class="lineno"> 722</span>  ARRAY[</div> |
| <div class="line"><a name="l00723"></a><span class="lineno"> 723</span>  encoded_table_name,</div> |
| <div class="line"><a name="l00724"></a><span class="lineno"> 724</span>  tr_table_name</div> |
| <div class="line"><a name="l00725"></a><span class="lineno"> 725</span>  ]</div> |
| <div class="line"><a name="l00726"></a><span class="lineno"> 726</span>  ); </div> |
| <div class="line"><a name="l00727"></a><span class="lineno"> 727</span>  END IF;</div> |
| <div class="line"><a name="l00728"></a><span class="lineno"> 728</span>  DROP TABLE IF EXISTS training_instance_aux;</div> |
| <div class="line"><a name="l00729"></a><span class="lineno"> 729</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l00730"></a><span class="lineno"> 730</span>  (</div> |
| <div class="line"><a name="l00731"></a><span class="lineno"> 731</span>  'CREATE TEMP TABLE training_instance_aux AS</div> |
| <div class="line"><a name="l00732"></a><span class="lineno"> 732</span>  SELECT tid, nid, fid, fval, is_cont, </div> |
| <div class="line"><a name="l00733"></a><span class="lineno"> 733</span>  MADLIB_SCHEMA.__dt_acc_count_aggr</div> |
| <div class="line"><a name="l00734"></a><span class="lineno"> 734</span>  (%,count::BIGINT,class::INT) AS count</div> |
| <div class="line"><a name="l00735"></a><span class="lineno"> 735</span>  FROM </div> |
| <div class="line"><a name="l00736"></a><span class="lineno"> 736</span>  (</div> |
| <div class="line"><a name="l00737"></a><span class="lineno"> 737</span>  % </div> |
| <div class="line"><a name="l00738"></a><span class="lineno"> 738</span>  ) l</div> |
| <div class="line"><a name="l00739"></a><span class="lineno"> 739</span>  GROUP BY tid,nid,fid, fval,is_cont </div> |
| <div class="line"><a name="l00740"></a><span class="lineno"> 740</span>  m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (fid, fval)')',</div> |
| <div class="line"><a name="l00741"></a><span class="lineno"> 741</span>  ARRAY[</div> |
| <div class="line"><a name="l00742"></a><span class="lineno"> 742</span>  num_classes::TEXT,</div> |
| <div class="line"><a name="l00743"></a><span class="lineno"> 743</span>  select_stmt</div> |
| <div class="line"><a name="l00744"></a><span class="lineno"> 744</span>  ]</div> |
| <div class="line"><a name="l00745"></a><span class="lineno"> 745</span>  );</div> |
| <div class="line"><a name="l00746"></a><span class="lineno"> 746</span> </div> |
| <div class="line"><a name="l00747"></a><span class="lineno"> 747</span>  IF ( verbosity>0 ) THEN</div> |
| <div class="line"><a name="l00748"></a><span class="lineno"> 748</span>  RAISE INFO '%', curstmt;</div> |
| <div class="line"><a name="l00749"></a><span class="lineno"> 749</span>  END IF;</div> |
| <div class="line"><a name="l00750"></a><span class="lineno"> 750</span>  </div> |
| <div class="line"><a name="l00751"></a><span class="lineno"> 751</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l00752"></a><span class="lineno"> 752</span>  ret.calc_acc_time = clock_timestamp() - begin_calc_acc;</div> |
| <div class="line"><a name="l00753"></a><span class="lineno"> 753</span>  </div> |
| <div class="line"><a name="l00754"></a><span class="lineno"> 754</span>  RETURN ret;</div> |
| <div class="line"><a name="l00755"></a><span class="lineno"> 755</span> END</div> |
| <div class="line"><a name="l00756"></a><span class="lineno"> 756</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l00757"></a><span class="lineno"> 757</span> </div> |
| <div class="line"><a name="l00758"></a><span class="lineno"> 758</span> </div> |
| <div class="line"><a name="l00759"></a><span class="lineno"> 759</span> DROP TYPE IF EXISTS MADLIB_SCHEMA.__rep_type CASCADE;</div> |
| <div class="line"><a name="l00760"></a><span class="lineno"> 760</span> CREATE TYPE MADLIB_SCHEMA.__rep_type AS</div> |
| <div class="line"><a name="l00761"></a><span class="lineno"> 761</span>  (</div> |
| <div class="line"><a name="l00762"></a><span class="lineno"> 762</span>  numOfOrgClasses BIGINT[]</div> |
| <div class="line"><a name="l00763"></a><span class="lineno"> 763</span>  );</div> |
| <div class="line"><a name="l00764"></a><span class="lineno"> 764</span> </div> |
| <div class="line"><a name="l00765"></a><span class="lineno"> 765</span> </div> |
| <div class="line"><a name="l00766"></a><span class="lineno"> 766</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00767"></a><span class="lineno"> 767</span> <span class="comment"> * @brief The step function for aggregating the class counts while doing Reduce </span></div> |
| <div class="line"><a name="l00768"></a><span class="lineno"> 768</span> <span class="comment"> * Error Pruning (REP).</span></div> |
| <div class="line"><a name="l00769"></a><span class="lineno"> 769</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00770"></a><span class="lineno"> 770</span> <span class="comment"> * @param class_count_array The array used to store the accumulated information.</span></div> |
| <div class="line"><a name="l00771"></a><span class="lineno"> 771</span> <span class="comment"> * [0]: the total number of mis-classified samples.</span></div> |
| <div class="line"><a name="l00772"></a><span class="lineno"> 772</span> <span class="comment"> * [i]: the number of samples belonging to the ith class.</span></div> |
| <div class="line"><a name="l00773"></a><span class="lineno"> 773</span> <span class="comment"> * @param classified_class The predicted class based on our trained DT model.</span></div> |
| <div class="line"><a name="l00774"></a><span class="lineno"> 774</span> <span class="comment"> * @param original_class The real class value provided in the validation set.</span></div> |
| <div class="line"><a name="l00775"></a><span class="lineno"> 775</span> <span class="comment"> * @param max_num_of_classes The total number of distinct class values. </span></div> |
| <div class="line"><a name="l00776"></a><span class="lineno"> 776</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00777"></a><span class="lineno"> 777</span> <span class="comment"> * @return An updated class count array.</span></div> |
| <div class="line"><a name="l00778"></a><span class="lineno"> 778</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00779"></a><span class="lineno"> 779</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00780"></a><span class="lineno"> 780</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__rep_aggr_class_count_sfunc</div> |
| <div class="line"><a name="l00781"></a><span class="lineno"> 781</span>  (</div> |
| <div class="line"><a name="l00782"></a><span class="lineno"> 782</span>  class_count_array BIGINT[], </div> |
| <div class="line"><a name="l00783"></a><span class="lineno"> 783</span>  classified_class INT,</div> |
| <div class="line"><a name="l00784"></a><span class="lineno"> 784</span>  original_class INT,</div> |
| <div class="line"><a name="l00785"></a><span class="lineno"> 785</span>  max_num_of_classes INT</div> |
| <div class="line"><a name="l00786"></a><span class="lineno"> 786</span>  ) </div> |
| <div class="line"><a name="l00787"></a><span class="lineno"> 787</span> RETURNS BIGINT[]</div> |
| <div class="line"><a name="l00788"></a><span class="lineno"> 788</span> AS 'MODULE_PATHNAME', 'dt_rep_aggr_class_count_sfunc'</div> |
| <div class="line"><a name="l00789"></a><span class="lineno"> 789</span> LANGUAGE C IMMUTABLE;</div> |
| <div class="line"><a name="l00790"></a><span class="lineno"> 790</span> </div> |
| <div class="line"><a name="l00791"></a><span class="lineno"> 791</span> </div> |
| <div class="line"><a name="l00792"></a><span class="lineno"> 792</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00793"></a><span class="lineno"> 793</span> <span class="comment"> * @brief Add the corresponding elements of the input arrays </span></div> |
| <div class="line"><a name="l00794"></a><span class="lineno"> 794</span> <span class="comment"> * to create a new one.</span></div> |
| <div class="line"><a name="l00795"></a><span class="lineno"> 795</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00796"></a><span class="lineno"> 796</span> <span class="comment"> * @param 1 arg The array 1.</span></div> |
| <div class="line"><a name="l00797"></a><span class="lineno"> 797</span> <span class="comment"> * @param 2 arg The array 2.</span></div> |
| <div class="line"><a name="l00798"></a><span class="lineno"> 798</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00799"></a><span class="lineno"> 799</span> <span class="comment"> * @return The new array.</span></div> |
| <div class="line"><a name="l00800"></a><span class="lineno"> 800</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00801"></a><span class="lineno"> 801</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00802"></a><span class="lineno"> 802</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__bigint_array_add</div> |
| <div class="line"><a name="l00803"></a><span class="lineno"> 803</span>  (</div> |
| <div class="line"><a name="l00804"></a><span class="lineno"> 804</span>  BIGINT[],</div> |
| <div class="line"><a name="l00805"></a><span class="lineno"> 805</span>  BIGINT[]</div> |
| <div class="line"><a name="l00806"></a><span class="lineno"> 806</span>  ) </div> |
| <div class="line"><a name="l00807"></a><span class="lineno"> 807</span> RETURNS BIGINT[]</div> |
| <div class="line"><a name="l00808"></a><span class="lineno"> 808</span> AS 'MODULE_PATHNAME', 'bigint_array_add'</div> |
| <div class="line"><a name="l00809"></a><span class="lineno"> 809</span> LANGUAGE C IMMUTABLE;</div> |
| <div class="line"><a name="l00810"></a><span class="lineno"> 810</span> </div> |
| <div class="line"><a name="l00811"></a><span class="lineno"> 811</span> </div> |
| <div class="line"><a name="l00812"></a><span class="lineno"> 812</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00813"></a><span class="lineno"> 813</span> <span class="comment"> * @brief The final function for aggregating the class counts for REP. </span></div> |
| <div class="line"><a name="l00814"></a><span class="lineno"> 814</span> <span class="comment"> * It takes the class count array produced by the sfunc and produces a </span></div> |
| <div class="line"><a name="l00815"></a><span class="lineno"> 815</span> <span class="comment"> * two-element array. The first element is the ID of the class that has </span></div> |
| <div class="line"><a name="l00816"></a><span class="lineno"> 816</span> <span class="comment"> * the maximum number of samples represented by the root node of the subtree</span></div> |
| <div class="line"><a name="l00817"></a><span class="lineno"> 817</span> <span class="comment"> * being processed. The second element is the number of reduced </span></div> |
| <div class="line"><a name="l00818"></a><span class="lineno"> 818</span> <span class="comment"> * misclassified samples if the leave nodes of the subtree are pruned.</span></div> |
| <div class="line"><a name="l00819"></a><span class="lineno"> 819</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00820"></a><span class="lineno"> 820</span> <span class="comment"> * @param class_count_data The array containing all the information for the </span></div> |
| <div class="line"><a name="l00821"></a><span class="lineno"> 821</span> <span class="comment"> * calculation of Reduced-Error pruning. </span></div> |
| <div class="line"><a name="l00822"></a><span class="lineno"> 822</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00823"></a><span class="lineno"> 823</span> <span class="comment"> * @return A two element array.</span></div> |
| <div class="line"><a name="l00824"></a><span class="lineno"> 824</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00825"></a><span class="lineno"> 825</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00826"></a><span class="lineno"> 826</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__rep_aggr_class_count_ffunc</div> |
| <div class="line"><a name="l00827"></a><span class="lineno"> 827</span>  (</div> |
| <div class="line"><a name="l00828"></a><span class="lineno"> 828</span>  class_count_array BIGINT[] </div> |
| <div class="line"><a name="l00829"></a><span class="lineno"> 829</span>  ) </div> |
| <div class="line"><a name="l00830"></a><span class="lineno"> 830</span> RETURNS BIGINT[]</div> |
| <div class="line"><a name="l00831"></a><span class="lineno"> 831</span> AS 'MODULE_PATHNAME', 'dt_rep_aggr_class_count_ffunc'</div> |
| <div class="line"><a name="l00832"></a><span class="lineno"> 832</span> LANGUAGE C STRICT IMMUTABLE;</div> |
| <div class="line"><a name="l00833"></a><span class="lineno"> 833</span> </div> |
| <div class="line"><a name="l00834"></a><span class="lineno"> 834</span> </div> |
| <div class="line"><a name="l00835"></a><span class="lineno"> 835</span> DROP AGGREGATE IF EXISTS MADLIB_SCHEMA.__rep_aggr_class_count</div> |
| <div class="line"><a name="l00836"></a><span class="lineno"> 836</span>  (</div> |
| <div class="line"><a name="l00837"></a><span class="lineno"> 837</span>  INT,</div> |
| <div class="line"><a name="l00838"></a><span class="lineno"> 838</span>  INT,</div> |
| <div class="line"><a name="l00839"></a><span class="lineno"> 839</span>  INT</div> |
| <div class="line"><a name="l00840"></a><span class="lineno"> 840</span>  );</div> |
| <div class="line"><a name="l00841"></a><span class="lineno"> 841</span> CREATE AGGREGATE MADLIB_SCHEMA.__rep_aggr_class_count</div> |
| <div class="line"><a name="l00842"></a><span class="lineno"> 842</span>  (</div> |
| <div class="line"><a name="l00843"></a><span class="lineno"> 843</span>  INT,</div> |
| <div class="line"><a name="l00844"></a><span class="lineno"> 844</span>  INT,</div> |
| <div class="line"><a name="l00845"></a><span class="lineno"> 845</span>  INT</div> |
| <div class="line"><a name="l00846"></a><span class="lineno"> 846</span>  ) </div> |
| <div class="line"><a name="l00847"></a><span class="lineno"> 847</span> (</div> |
| <div class="line"><a name="l00848"></a><span class="lineno"> 848</span>  SFUNC=MADLIB_SCHEMA.__rep_aggr_class_count_sfunc,</div> |
| <div class="line"><a name="l00849"></a><span class="lineno"> 849</span>  m4_ifdef(`__GREENPLUM__', `prefunc=MADLIB_SCHEMA.__bigint_array_add,')</div> |
| <div class="line"><a name="l00850"></a><span class="lineno"> 850</span>  FINALFUNC=MADLIB_SCHEMA.__rep_aggr_class_count_ffunc,</div> |
| <div class="line"><a name="l00851"></a><span class="lineno"> 851</span>  STYPE=BIGINT[]</div> |
| <div class="line"><a name="l00852"></a><span class="lineno"> 852</span> );</div> |
| <div class="line"><a name="l00853"></a><span class="lineno"> 853</span> </div> |
| <div class="line"><a name="l00854"></a><span class="lineno"> 854</span> </div> |
| <div class="line"><a name="l00855"></a><span class="lineno"> 855</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00856"></a><span class="lineno"> 856</span> <span class="comment"> * @brief The step function of the aggregate __array_indexed_agg.</span></div> |
| <div class="line"><a name="l00857"></a><span class="lineno"> 857</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00858"></a><span class="lineno"> 858</span> <span class="comment"> * @param state The step state array of the aggregate function.</span></div> |
| <div class="line"><a name="l00859"></a><span class="lineno"> 859</span> <span class="comment"> * @param elem The element to be filled into the state array.</span></div> |
| <div class="line"><a name="l00860"></a><span class="lineno"> 860</span> <span class="comment"> * @param elem_cnt The number of elements.</span></div> |
| <div class="line"><a name="l00861"></a><span class="lineno"> 861</span> <span class="comment"> * @param elem_idx the subscript of "elem" in the state array.</span></div> |
| <div class="line"><a name="l00862"></a><span class="lineno"> 862</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00863"></a><span class="lineno"> 863</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00864"></a><span class="lineno"> 864</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__array_indexed_agg_sfunc</div> |
| <div class="line"><a name="l00865"></a><span class="lineno"> 865</span>  (</div> |
| <div class="line"><a name="l00866"></a><span class="lineno"> 866</span>  state float8[], </div> |
| <div class="line"><a name="l00867"></a><span class="lineno"> 867</span>  elem float8, </div> |
| <div class="line"><a name="l00868"></a><span class="lineno"> 868</span>  elem_cnt int8, </div> |
| <div class="line"><a name="l00869"></a><span class="lineno"> 869</span>  elem_idx int8</div> |
| <div class="line"><a name="l00870"></a><span class="lineno"> 870</span>  )</div> |
| <div class="line"><a name="l00871"></a><span class="lineno"> 871</span> RETURNS float8[]</div> |
| <div class="line"><a name="l00872"></a><span class="lineno"> 872</span> AS 'MODULE_PATHNAME', 'dt_array_indexed_agg_sfunc' </div> |
| <div class="line"><a name="l00873"></a><span class="lineno"> 873</span> LANGUAGE C IMMUTABLE;</div> |
| <div class="line"><a name="l00874"></a><span class="lineno"> 874</span> </div> |
| <div class="line"><a name="l00875"></a><span class="lineno"> 875</span> </div> |
| <div class="line"><a name="l00876"></a><span class="lineno"> 876</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00877"></a><span class="lineno"> 877</span> <span class="comment"> * @brief The Pre-function of the aggregate __array_indexed_agg.</span></div> |
| <div class="line"><a name="l00878"></a><span class="lineno"> 878</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00879"></a><span class="lineno"> 879</span> <span class="comment"> * @param arg0 The first state array.</span></div> |
| <div class="line"><a name="l00880"></a><span class="lineno"> 880</span> <span class="comment"> * @param arg1 The second state array.</span></div> |
| <div class="line"><a name="l00881"></a><span class="lineno"> 881</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00882"></a><span class="lineno"> 882</span> <span class="comment"> * @return The combined state. </span></div> |
| <div class="line"><a name="l00883"></a><span class="lineno"> 883</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00884"></a><span class="lineno"> 884</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00885"></a><span class="lineno"> 885</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__array_indexed_agg_prefunc</div> |
| <div class="line"><a name="l00886"></a><span class="lineno"> 886</span>  (</div> |
| <div class="line"><a name="l00887"></a><span class="lineno"> 887</span>  float8[], </div> |
| <div class="line"><a name="l00888"></a><span class="lineno"> 888</span>  float8[]</div> |
| <div class="line"><a name="l00889"></a><span class="lineno"> 889</span>  )</div> |
| <div class="line"><a name="l00890"></a><span class="lineno"> 890</span> RETURNS float8[]</div> |
| <div class="line"><a name="l00891"></a><span class="lineno"> 891</span> AS 'MODULE_PATHNAME', 'dt_array_indexed_agg_prefunc' </div> |
| <div class="line"><a name="l00892"></a><span class="lineno"> 892</span> LANGUAGE C STRICT IMMUTABLE;</div> |
| <div class="line"><a name="l00893"></a><span class="lineno"> 893</span> </div> |
| <div class="line"><a name="l00894"></a><span class="lineno"> 894</span> </div> |
| <div class="line"><a name="l00895"></a><span class="lineno"> 895</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00896"></a><span class="lineno"> 896</span> <span class="comment"> * @brief The final function of __array_indexed_agg.</span></div> |
| <div class="line"><a name="l00897"></a><span class="lineno"> 897</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00898"></a><span class="lineno"> 898</span> <span class="comment"> * @param state The state array.</span></div> |
| <div class="line"><a name="l00899"></a><span class="lineno"> 899</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00900"></a><span class="lineno"> 900</span> <span class="comment"> * @return The aggregate result.</span></div> |
| <div class="line"><a name="l00901"></a><span class="lineno"> 901</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00902"></a><span class="lineno"> 902</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00903"></a><span class="lineno"> 903</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__array_indexed_agg_ffunc</div> |
| <div class="line"><a name="l00904"></a><span class="lineno"> 904</span>  (</div> |
| <div class="line"><a name="l00905"></a><span class="lineno"> 905</span>  float8[]</div> |
| <div class="line"><a name="l00906"></a><span class="lineno"> 906</span>  )</div> |
| <div class="line"><a name="l00907"></a><span class="lineno"> 907</span> RETURNS float8[]</div> |
| <div class="line"><a name="l00908"></a><span class="lineno"> 908</span> AS 'MODULE_PATHNAME', 'dt_array_indexed_agg_ffunc' </div> |
| <div class="line"><a name="l00909"></a><span class="lineno"> 909</span> LANGUAGE C IMMUTABLE;</div> |
| <div class="line"><a name="l00910"></a><span class="lineno"> 910</span> </div> |
| <div class="line"><a name="l00911"></a><span class="lineno"> 911</span> </div> |
| <div class="line"><a name="l00912"></a><span class="lineno"> 912</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00913"></a><span class="lineno"> 913</span> <span class="comment"> * @brief The aggregate is the same with array_agg, which will accumulate</span></div> |
| <div class="line"><a name="l00914"></a><span class="lineno"> 914</span> <span class="comment"> * The elements in each group to an array, except that we allow users </span></div> |
| <div class="line"><a name="l00915"></a><span class="lineno"> 915</span> <span class="comment"> * provide the subscript for each element. This aggregate will be </span></div> |
| <div class="line"><a name="l00916"></a><span class="lineno"> 916</span> <span class="comment"> * invoked as HashAggregate, while array_agg will be called as </span></div> |
| <div class="line"><a name="l00917"></a><span class="lineno"> 917</span> <span class="comment"> * GroupAggregate. Therefore, our implementation have better performance</span></div> |
| <div class="line"><a name="l00918"></a><span class="lineno"> 918</span> <span class="comment"> * than the array_agg.</span></div> |
| <div class="line"><a name="l00919"></a><span class="lineno"> 919</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00920"></a><span class="lineno"> 920</span> <span class="comment"> * @param elem The element to be fed into the returned array of this aggregate.</span></div> |
| <div class="line"><a name="l00921"></a><span class="lineno"> 921</span> <span class="comment"> * @param elem_cnt The number of elements.</span></div> |
| <div class="line"><a name="l00922"></a><span class="lineno"> 922</span> <span class="comment"> * @param elem_idx The subscript of the element.</span></div> |
| <div class="line"><a name="l00923"></a><span class="lineno"> 923</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00924"></a><span class="lineno"> 924</span> <span class="comment"> * @return The aggregated array.</span></div> |
| <div class="line"><a name="l00925"></a><span class="lineno"> 925</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00926"></a><span class="lineno"> 926</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00927"></a><span class="lineno"> 927</span> CREATE AGGREGATE MADLIB_SCHEMA.__array_indexed_agg(float8, int8, int8) (</div> |
| <div class="line"><a name="l00928"></a><span class="lineno"> 928</span>  SFUNC = MADLIB_SCHEMA.__array_indexed_agg_sfunc,</div> |
| <div class="line"><a name="l00929"></a><span class="lineno"> 929</span>  m4_ifdef( `__GREENPLUM__',`PREFUNC = MADLIB_SCHEMA.__array_indexed_agg_prefunc,')</div> |
| <div class="line"><a name="l00930"></a><span class="lineno"> 930</span>  FINALFUNC = MADLIB_SCHEMA.__array_indexed_agg_ffunc, </div> |
| <div class="line"><a name="l00931"></a><span class="lineno"> 931</span>  STYPE = float8[]</div> |
| <div class="line"><a name="l00932"></a><span class="lineno"> 932</span> );</div> |
| <div class="line"><a name="l00933"></a><span class="lineno"> 933</span> </div> |
| <div class="line"><a name="l00934"></a><span class="lineno"> 934</span> </div> |
| <div class="line"><a name="l00935"></a><span class="lineno"> 935</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__dt_acc_count_sfunc</div> |
| <div class="line"><a name="l00936"></a><span class="lineno"> 936</span>  (</div> |
| <div class="line"><a name="l00937"></a><span class="lineno"> 937</span>  count_array BIGINT[],</div> |
| <div class="line"><a name="l00938"></a><span class="lineno"> 938</span>  num_of_class INT,</div> |
| <div class="line"><a name="l00939"></a><span class="lineno"> 939</span>  count BIGINT,</div> |
| <div class="line"><a name="l00940"></a><span class="lineno"> 940</span>  class INT</div> |
| <div class="line"><a name="l00941"></a><span class="lineno"> 941</span>  ) </div> |
| <div class="line"><a name="l00942"></a><span class="lineno"> 942</span> RETURNS BIGINT[] </div> |
| <div class="line"><a name="l00943"></a><span class="lineno"> 943</span> AS 'MODULE_PATHNAME', 'dt_acc_count_sfunc'</div> |
| <div class="line"><a name="l00944"></a><span class="lineno"> 944</span> LANGUAGE C VOLATILE;</div> |
| <div class="line"><a name="l00945"></a><span class="lineno"> 945</span> </div> |
| <div class="line"><a name="l00946"></a><span class="lineno"> 946</span> </div> |
| <div class="line"><a name="l00947"></a><span class="lineno"> 947</span> CREATE AGGREGATE MADLIB_SCHEMA.__dt_acc_count_aggr</div> |
| <div class="line"><a name="l00948"></a><span class="lineno"> 948</span>  (</div> |
| <div class="line"><a name="l00949"></a><span class="lineno"> 949</span>  INT,</div> |
| <div class="line"><a name="l00950"></a><span class="lineno"> 950</span>  BIGINT,</div> |
| <div class="line"><a name="l00951"></a><span class="lineno"> 951</span>  INT</div> |
| <div class="line"><a name="l00952"></a><span class="lineno"> 952</span>  ) </div> |
| <div class="line"><a name="l00953"></a><span class="lineno"> 953</span> (</div> |
| <div class="line"><a name="l00954"></a><span class="lineno"> 954</span>  SFUNC=MADLIB_SCHEMA.__dt_acc_count_sfunc,</div> |
| <div class="line"><a name="l00955"></a><span class="lineno"> 955</span>  m4_ifdef(`__GREENPLUM__', `prefunc=MADLIB_SCHEMA.__bigint_array_add,')</div> |
| <div class="line"><a name="l00956"></a><span class="lineno"> 956</span>  STYPE=BIGINT[]</div> |
| <div class="line"><a name="l00957"></a><span class="lineno"> 957</span> );</div> |
| <div class="line"><a name="l00958"></a><span class="lineno"> 958</span> </div> |
| <div class="line"><a name="l00959"></a><span class="lineno"> 959</span> </div> |
| <div class="line"><a name="l00960"></a><span class="lineno"> 960</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00961"></a><span class="lineno"> 961</span> <span class="comment"> * @brief The aggregate is created for the PostgreSQL, which doesn't support the</span></div> |
| <div class="line"><a name="l00962"></a><span class="lineno"> 962</span> <span class="comment"> * function sum over an array.</span></div> |
| <div class="line"><a name="l00963"></a><span class="lineno"> 963</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l00964"></a><span class="lineno"> 964</span> <span class="comment"> * @param elem The element to be fed into the returned array of this aggregate.</span></div> |
| <div class="line"><a name="l00965"></a><span class="lineno"> 965</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00966"></a><span class="lineno"> 966</span> <span class="comment"> * @return The array with the sum of all the input array in a group.</span></div> |
| <div class="line"><a name="l00967"></a><span class="lineno"> 967</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00968"></a><span class="lineno"> 968</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00969"></a><span class="lineno"> 969</span> CREATE</div> |
| <div class="line"><a name="l00970"></a><span class="lineno"> 970</span> AGGREGATE MADLIB_SCHEMA.__bigint_array_sum</div> |
| <div class="line"><a name="l00971"></a><span class="lineno"> 971</span>  (</div> |
| <div class="line"><a name="l00972"></a><span class="lineno"> 972</span>  BIGINT[]</div> |
| <div class="line"><a name="l00973"></a><span class="lineno"> 973</span>  ) </div> |
| <div class="line"><a name="l00974"></a><span class="lineno"> 974</span> (</div> |
| <div class="line"><a name="l00975"></a><span class="lineno"> 975</span>  SFUNC=MADLIB_SCHEMA.__bigint_array_add,</div> |
| <div class="line"><a name="l00976"></a><span class="lineno"> 976</span>  m4_ifdef(`__GREENPLUM__', `prefunc=MADLIB_SCHEMA.__bigint_array_add,')</div> |
| <div class="line"><a name="l00977"></a><span class="lineno"> 977</span>  STYPE=BIGINT[]</div> |
| <div class="line"><a name="l00978"></a><span class="lineno"> 978</span> );</div> |
| <div class="line"><a name="l00979"></a><span class="lineno"> 979</span> </div> |
| <div class="line"><a name="l00980"></a><span class="lineno"> 980</span> </div> |
| <div class="line"><a name="l00981"></a><span class="lineno"> 981</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00982"></a><span class="lineno"> 982</span> <span class="comment"> * @brief This function find the best split and return the information.</span></div> |
| <div class="line"><a name="l00983"></a><span class="lineno"> 983</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00984"></a><span class="lineno"> 984</span> <span class="comment"> * @param table_name The name of the table containing the training</span></div> |
| <div class="line"><a name="l00985"></a><span class="lineno"> 985</span> <span class="comment"> * set.</span></div> |
| <div class="line"><a name="l00986"></a><span class="lineno"> 986</span> <span class="comment"> * @param confidence_level This parameter is used by the 'Error-Based Pruning'.</span></div> |
| <div class="line"><a name="l00987"></a><span class="lineno"> 987</span> <span class="comment"> * Please refer to the paper for detailed definition.</span></div> |
| <div class="line"><a name="l00988"></a><span class="lineno"> 988</span> <span class="comment"> * The paper's name is 'Error-Based Pruning of Decision </span></div> |
| <div class="line"><a name="l00989"></a><span class="lineno"> 989</span> <span class="comment"> * Trees Grown on Very Large Data Sets Can Work!'.</span></div> |
| <div class="line"><a name="l00990"></a><span class="lineno"> 990</span> <span class="comment"> * @param feature_table_name Is is the name of one internal table, which contains</span></div> |
| <div class="line"><a name="l00991"></a><span class="lineno"> 991</span> <span class="comment"> * meta data for each feature.</span></div> |
| <div class="line"><a name="l00992"></a><span class="lineno"> 992</span> <span class="comment"> * @param split_criterion It defines the split criterion to be used.</span></div> |
| <div class="line"><a name="l00993"></a><span class="lineno"> 993</span> <span class="comment"> * (1- information gain. 2- gain ratio. 3- gini).</span></div> |
| <div class="line"><a name="l00994"></a><span class="lineno"> 994</span> <span class="comment"> * @param continue_grow It specifies whether we should still grow the tree</span></div> |
| <div class="line"><a name="l00995"></a><span class="lineno"> 995</span> <span class="comment"> * on the selected branch.</span></div> |
| <div class="line"><a name="l00996"></a><span class="lineno"> 996</span> <span class="comment"> * @param output_table It specifies the table used to store the chosen splits.</span></div> |
| <div class="line"><a name="l00997"></a><span class="lineno"> 997</span> <span class="comment"> * @param h2hmv_routine_id Specifies how to handle missing values. </span></div> |
| <div class="line"><a name="l00998"></a><span class="lineno"> 998</span> <span class="comment"> * 1 ignore, 2 explicit.</span></div> |
| <div class="line"><a name="l00999"></a><span class="lineno"> 999</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l01000"></a><span class="lineno"> 1000</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l01001"></a><span class="lineno"> 1001</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__find_best_split</div> |
| <div class="line"><a name="l01002"></a><span class="lineno"> 1002</span>  (</div> |
| <div class="line"><a name="l01003"></a><span class="lineno"> 1003</span>  table_name TEXT, </div> |
| <div class="line"><a name="l01004"></a><span class="lineno"> 1004</span>  confidence_level FLOAT,</div> |
| <div class="line"><a name="l01005"></a><span class="lineno"> 1005</span>  feature_table_name TEXT, </div> |
| <div class="line"><a name="l01006"></a><span class="lineno"> 1006</span>  split_criterion INT, </div> |
| <div class="line"><a name="l01007"></a><span class="lineno"> 1007</span>  continue_grow INT,</div> |
| <div class="line"><a name="l01008"></a><span class="lineno"> 1008</span>  output_table TEXT,</div> |
| <div class="line"><a name="l01009"></a><span class="lineno"> 1009</span>  h2hmv_routine_id INT,</div> |
| <div class="line"><a name="l01010"></a><span class="lineno"> 1010</span>  num_classes INT</div> |
| <div class="line"><a name="l01011"></a><span class="lineno"> 1011</span>  ) </div> |
| <div class="line"><a name="l01012"></a><span class="lineno"> 1012</span> RETURNS VOID AS $$</div> |
| <div class="line"><a name="l01013"></a><span class="lineno"> 1013</span> DECLARE</div> |
| <div class="line"><a name="l01014"></a><span class="lineno"> 1014</span>  total_size INT;</div> |
| <div class="line"><a name="l01015"></a><span class="lineno"> 1015</span>  curstmt TEXT := '';</div> |
| <div class="line"><a name="l01016"></a><span class="lineno"> 1016</span>  begin_func_exec TIMESTAMP;</div> |
| <div class="line"><a name="l01017"></a><span class="lineno"> 1017</span>  select_stmt TEXT;</div> |
| <div class="line"><a name="l01018"></a><span class="lineno"> 1018</span> BEGIN </div> |
| <div class="line"><a name="l01019"></a><span class="lineno"> 1019</span>  begin_func_exec = clock_timestamp();</div> |
| <div class="line"><a name="l01020"></a><span class="lineno"> 1020</span>  </div> |
| <div class="line"><a name="l01021"></a><span class="lineno"> 1021</span>  IF (h2hmv_routine_id=1) THEN</div> |
| <div class="line"><a name="l01022"></a><span class="lineno"> 1022</span>  -- For ignore, we need the true size of nodes to handle the missing values.</div> |
| <div class="line"><a name="l01023"></a><span class="lineno"> 1023</span>  select_stmt = </div> |
| <div class="line"><a name="l01024"></a><span class="lineno"> 1024</span>  'SELECT t1.tid, t1.nid, t1.fid, t1.total, t2.node_size::BIGINT </div> |
| <div class="line"><a name="l01025"></a><span class="lineno"> 1025</span>  FROM</div> |
| <div class="line"><a name="l01026"></a><span class="lineno"> 1026</span>  (</div> |
| <div class="line"><a name="l01027"></a><span class="lineno"> 1027</span>  SELECT tid, nid, fid, </div> |
| <div class="line"><a name="l01028"></a><span class="lineno"> 1028</span>  m4_ifdef(`__GREENPLUM__', `sum(count)', `MADLIB_SCHEMA.__bigint_array_sum(count)') as total</div> |
| <div class="line"><a name="l01029"></a><span class="lineno"> 1029</span>  FROM training_instance_aux</div> |
| <div class="line"><a name="l01030"></a><span class="lineno"> 1030</span>  GROUP BY tid, nid, fid</div> |
| <div class="line"><a name="l01031"></a><span class="lineno"> 1031</span>  ) t1 INNER JOIN node_size_aux t2 </div> |
| <div class="line"><a name="l01032"></a><span class="lineno"> 1032</span>  ON t1.tid=t2.tid AND t1.nid=t2.nid';</div> |
| <div class="line"><a name="l01033"></a><span class="lineno"> 1033</span>  ELSE</div> |
| <div class="line"><a name="l01034"></a><span class="lineno"> 1034</span>  -- For explicit, the calculated node size from the aggregation is correct.</div> |
| <div class="line"><a name="l01035"></a><span class="lineno"> 1035</span>  -- We can set NULL, which denotes we can safely use the counted value.</div> |
| <div class="line"><a name="l01036"></a><span class="lineno"> 1036</span>  select_stmt = </div> |
| <div class="line"><a name="l01037"></a><span class="lineno"> 1037</span>  'SELECT tid, nid, fid, </div> |
| <div class="line"><a name="l01038"></a><span class="lineno"> 1038</span>  m4_ifdef(`__GREENPLUM__', `sum(count)', `MADLIB_SCHEMA.__bigint_array_sum(count)') as total, </div> |
| <div class="line"><a name="l01039"></a><span class="lineno"> 1039</span>  NULL::BIGINT AS node_size</div> |
| <div class="line"><a name="l01040"></a><span class="lineno"> 1040</span>  FROM training_instance_aux</div> |
| <div class="line"><a name="l01041"></a><span class="lineno"> 1041</span>  GROUP BY tid, nid, fid';</div> |
| <div class="line"><a name="l01042"></a><span class="lineno"> 1042</span>  END IF;</div> |
| <div class="line"><a name="l01043"></a><span class="lineno"> 1043</span> </div> |
| <div class="line"><a name="l01044"></a><span class="lineno"> 1044</span>  <span class="comment">/*</span></div> |
| <div class="line"><a name="l01045"></a><span class="lineno"> 1045</span> <span class="comment"> * This table is used to store information for the calculated best split </span></div> |
| <div class="line"><a name="l01046"></a><span class="lineno"> 1046</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01047"></a><span class="lineno"> 1047</span> <span class="comment"> * tid The ID of the tree.</span></div> |
| <div class="line"><a name="l01048"></a><span class="lineno"> 1048</span> <span class="comment"> * node_id The ID of one node in the specified tree.</span></div> |
| <div class="line"><a name="l01049"></a><span class="lineno"> 1049</span> <span class="comment"> * feature The ID of the selected feature.</span></div> |
| <div class="line"><a name="l01050"></a><span class="lineno"> 1050</span> <span class="comment"> * probability The predicted probability of our chosen class.</span></div> |
| <div class="line"><a name="l01051"></a><span class="lineno"> 1051</span> <span class="comment"> * max_class The ID of the class chosen by the algorithm.</span></div> |
| <div class="line"><a name="l01052"></a><span class="lineno"> 1052</span> <span class="comment"> * max_scv The maximum split criterion value.</span></div> |
| <div class="line"><a name="l01053"></a><span class="lineno"> 1053</span> <span class="comment"> * live 1- For the chosen split, we should split further.</span></div> |
| <div class="line"><a name="l01054"></a><span class="lineno"> 1054</span> <span class="comment"> * 0- For the chosen split, we shouldn't split further.</span></div> |
| <div class="line"><a name="l01055"></a><span class="lineno"> 1055</span> <span class="comment"> * ebp_coeff total error for error-based pruning.</span></div> |
| <div class="line"><a name="l01056"></a><span class="lineno"> 1056</span> <span class="comment"> * is_cont whether the selected feature is continuous.</span></div> |
| <div class="line"><a name="l01057"></a><span class="lineno"> 1057</span> <span class="comment"> * split_value If the selected feature is continuous, it specifies</span></div> |
| <div class="line"><a name="l01058"></a><span class="lineno"> 1058</span> <span class="comment"> * the split value. Otherwise, it is of no use.</span></div> |
| <div class="line"><a name="l01059"></a><span class="lineno"> 1059</span> <span class="comment"> * distinct_features The number of distinct values for the selected feature.</span></div> |
| <div class="line"><a name="l01060"></a><span class="lineno"> 1060</span> <span class="comment"> * node_size The size of this tree node. </span></div> |
| <div class="line"><a name="l01061"></a><span class="lineno"> 1061</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01062"></a><span class="lineno"> 1062</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l01063"></a><span class="lineno"> 1063</span>  EXECUTE 'DROP TABLE IF EXISTS '||output_table;</div> |
| <div class="line"><a name="l01064"></a><span class="lineno"> 1064</span>  EXECUTE 'CREATE TEMP TABLE '||output_table||' </div> |
| <div class="line"><a name="l01065"></a><span class="lineno"> 1065</span>  (</div> |
| <div class="line"><a name="l01066"></a><span class="lineno"> 1066</span>  tid INT,</div> |
| <div class="line"><a name="l01067"></a><span class="lineno"> 1067</span>  node_id INT,</div> |
| <div class="line"><a name="l01068"></a><span class="lineno"> 1068</span>  feature INT,</div> |
| <div class="line"><a name="l01069"></a><span class="lineno"> 1069</span>  probability FLOAT,</div> |
| <div class="line"><a name="l01070"></a><span class="lineno"> 1070</span>  max_class INTEGER,</div> |
| <div class="line"><a name="l01071"></a><span class="lineno"> 1071</span>  max_scv FLOAT,</div> |
| <div class="line"><a name="l01072"></a><span class="lineno"> 1072</span>  live INT,</div> |
| <div class="line"><a name="l01073"></a><span class="lineno"> 1073</span>  ebp_coeff FLOAT,</div> |
| <div class="line"><a name="l01074"></a><span class="lineno"> 1074</span>  is_cont BOOLEAN,</div> |
| <div class="line"><a name="l01075"></a><span class="lineno"> 1075</span>  split_value FLOAT,</div> |
| <div class="line"><a name="l01076"></a><span class="lineno"> 1076</span>  distinct_features INT,</div> |
| <div class="line"><a name="l01077"></a><span class="lineno"> 1077</span>  node_size INT</div> |
| <div class="line"><a name="l01078"></a><span class="lineno"> 1078</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (node_id)');';</div> |
| <div class="line"><a name="l01079"></a><span class="lineno"> 1079</span>  </div> |
| <div class="line"><a name="l01080"></a><span class="lineno"> 1080</span> </div> |
| <div class="line"><a name="l01081"></a><span class="lineno"> 1081</span>  EXECUTE 'DROP TABLE IF EXISTS tmp_best_table';</div> |
| <div class="line"><a name="l01082"></a><span class="lineno"> 1082</span> </div> |
| <div class="line"><a name="l01083"></a><span class="lineno"> 1083</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01084"></a><span class="lineno"> 1084</span>  (</div> |
| <div class="line"><a name="l01085"></a><span class="lineno"> 1085</span>  'INSERT INTO %</div> |
| <div class="line"><a name="l01086"></a><span class="lineno"> 1086</span>  SELECT tid, nid, best_scv[6], best_scv[4], best_scv[3], best_scv[1],</div> |
| <div class="line"><a name="l01087"></a><span class="lineno"> 1087</span>  CASE WHEN (best_scv[1] < 1e-9 OR </div> |
| <div class="line"><a name="l01088"></a><span class="lineno"> 1088</span>  best_scv[4] > 1-1e-9 OR % <= 0 ) THEN</div> |
| <div class="line"><a name="l01089"></a><span class="lineno"> 1089</span>  0</div> |
| <div class="line"><a name="l01090"></a><span class="lineno"> 1090</span>  ELSE</div> |
| <div class="line"><a name="l01091"></a><span class="lineno"> 1091</span>  1</div> |
| <div class="line"><a name="l01092"></a><span class="lineno"> 1092</span>  END AS live,</div> |
| <div class="line"><a name="l01093"></a><span class="lineno"> 1093</span>  MADLIB_SCHEMA.__ebp_calc_errors</div> |
| <div class="line"><a name="l01094"></a><span class="lineno"> 1094</span>  (best_scv[5], best_scv[4], %) AS ebp_coeff,</div> |
| <div class="line"><a name="l01095"></a><span class="lineno"> 1095</span>  o2.is_cont, </div> |
| <div class="line"><a name="l01096"></a><span class="lineno"> 1096</span>  CASE WHEN( o2.is_cont ) THEN</div> |
| <div class="line"><a name="l01097"></a><span class="lineno"> 1097</span>  best_scv[7]</div> |
| <div class="line"><a name="l01098"></a><span class="lineno"> 1098</span>  ELSE</div> |
| <div class="line"><a name="l01099"></a><span class="lineno"> 1099</span>  NULL</div> |
| <div class="line"><a name="l01100"></a><span class="lineno"> 1100</span>  END AS split_value,</div> |
| <div class="line"><a name="l01101"></a><span class="lineno"> 1101</span>  o2.num_dist_value, best_scv[5]</div> |
| <div class="line"><a name="l01102"></a><span class="lineno"> 1102</span>  FROM</div> |
| <div class="line"><a name="l01103"></a><span class="lineno"> 1103</span>  (</div> |
| <div class="line"><a name="l01104"></a><span class="lineno"> 1104</span>  SELECT s1.tid, s1.nid, </div> |
| <div class="line"><a name="l01105"></a><span class="lineno"> 1105</span>  MADLIB_SCHEMA.__best_scv_aggr(scv, s1.fid, </div> |
| <div class="line"><a name="l01106"></a><span class="lineno"> 1106</span>  coalesce(s1.split_value,0)) as best_scv</div> |
| <div class="line"><a name="l01107"></a><span class="lineno"> 1107</span>  FROM (</div> |
| <div class="line"><a name="l01108"></a><span class="lineno"> 1108</span>  SELECT t1.tid, t1.nid, t1.fid, split_value,</div> |
| <div class="line"><a name="l01109"></a><span class="lineno"> 1109</span>  MADLIB_SCHEMA.__scv_aggr</div> |
| <div class="line"><a name="l01110"></a><span class="lineno"> 1110</span>  (%, is_cont, %, le, total, t2.node_size) AS scv</div> |
| <div class="line"><a name="l01111"></a><span class="lineno"> 1111</span>  FROM </div> |
| <div class="line"><a name="l01112"></a><span class="lineno"> 1112</span>  (</div> |
| <div class="line"><a name="l01113"></a><span class="lineno"> 1113</span>  SELECT tid, nid, fid, fval, is_cont, </div> |
| <div class="line"><a name="l01114"></a><span class="lineno"> 1114</span>  CASE WHEN (is_cont) THEN</div> |
| <div class="line"><a name="l01115"></a><span class="lineno"> 1115</span>  fval </div> |
| <div class="line"><a name="l01116"></a><span class="lineno"> 1116</span>  ELSE</div> |
| <div class="line"><a name="l01117"></a><span class="lineno"> 1117</span>  NULL::FLOAT8</div> |
| <div class="line"><a name="l01118"></a><span class="lineno"> 1118</span>  END AS split_value,</div> |
| <div class="line"><a name="l01119"></a><span class="lineno"> 1119</span>  CASE WHEN (is_cont) THEN</div> |
| <div class="line"><a name="l01120"></a><span class="lineno"> 1120</span>  m4_ifdef(`__GREENPLUM__', `sum(count)', `MADLIB_SCHEMA.__bigint_array_sum(count)') OVER </div> |
| <div class="line"><a name="l01121"></a><span class="lineno"> 1121</span>  (PARTITION BY tid, nid, fid ORDER BY fval</div> |
| <div class="line"><a name="l01122"></a><span class="lineno"> 1122</span>  ROWS BETWEEN UNBOUNDED PRECEDING AND CURRENT ROW)</div> |
| <div class="line"><a name="l01123"></a><span class="lineno"> 1123</span>  ELSE</div> |
| <div class="line"><a name="l01124"></a><span class="lineno"> 1124</span>  count</div> |
| <div class="line"><a name="l01125"></a><span class="lineno"> 1125</span>  END AS le</div> |
| <div class="line"><a name="l01126"></a><span class="lineno"> 1126</span>  FROM training_instance_aux</div> |
| <div class="line"><a name="l01127"></a><span class="lineno"> 1127</span>  ) t1,</div> |
| <div class="line"><a name="l01128"></a><span class="lineno"> 1128</span>  (</div> |
| <div class="line"><a name="l01129"></a><span class="lineno"> 1129</span>  %</div> |
| <div class="line"><a name="l01130"></a><span class="lineno"> 1130</span>  ) t2</div> |
| <div class="line"><a name="l01131"></a><span class="lineno"> 1131</span>  WHERE t1.tid = t2.tid AND t1.nid = t2.nid AND t1.fid = t2.fid</div> |
| <div class="line"><a name="l01132"></a><span class="lineno"> 1132</span>  GROUP BY t1.tid, t1.nid, t1.fid, split_value</div> |
| <div class="line"><a name="l01133"></a><span class="lineno"> 1133</span>  ) s1</div> |
| <div class="line"><a name="l01134"></a><span class="lineno"> 1134</span>  GROUP BY s1.tid, s1.nid</div> |
| <div class="line"><a name="l01135"></a><span class="lineno"> 1135</span>  ) o1 INNER JOIN % o2 ON o1.best_scv[6]::INT=o2.<span class="keywordtype">id</span>',</div> |
| <div class="line"><a name="l01136"></a><span class="lineno"> 1136</span>  ARRAY[</div> |
| <div class="line"><a name="l01137"></a><span class="lineno"> 1137</span>  output_table,</div> |
| <div class="line"><a name="l01138"></a><span class="lineno"> 1138</span>  continue_grow::TEXT,</div> |
| <div class="line"><a name="l01139"></a><span class="lineno"> 1139</span>  confidence_level::TEXT, </div> |
| <div class="line"><a name="l01140"></a><span class="lineno"> 1140</span>  split_criterion::TEXT,</div> |
| <div class="line"><a name="l01141"></a><span class="lineno"> 1141</span>  num_classes::TEXT,</div> |
| <div class="line"><a name="l01142"></a><span class="lineno"> 1142</span>  select_stmt,</div> |
| <div class="line"><a name="l01143"></a><span class="lineno"> 1143</span>  feature_table_name</div> |
| <div class="line"><a name="l01144"></a><span class="lineno"> 1144</span>  ]</div> |
| <div class="line"><a name="l01145"></a><span class="lineno"> 1145</span>  ) INTO curstmt;</div> |
| <div class="line"><a name="l01146"></a><span class="lineno"> 1146</span> </div> |
| <div class="line"><a name="l01147"></a><span class="lineno"> 1147</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l01148"></a><span class="lineno"> 1148</span>  </div> |
| <div class="line"><a name="l01149"></a><span class="lineno"> 1149</span>  RETURN;</div> |
| <div class="line"><a name="l01150"></a><span class="lineno"> 1150</span> END</div> |
| <div class="line"><a name="l01151"></a><span class="lineno"> 1151</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l01152"></a><span class="lineno"> 1152</span> </div> |
| <div class="line"><a name="l01153"></a><span class="lineno"> 1153</span> </div> |
| <div class="line"><a name="l01154"></a><span class="lineno"> 1154</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l01155"></a><span class="lineno"> 1155</span> <span class="comment"> * @brief For training one decision tree, we need some internal tables</span></div> |
| <div class="line"><a name="l01156"></a><span class="lineno"> 1156</span> <span class="comment"> * to store intermediate results. This function creates those</span></div> |
| <div class="line"><a name="l01157"></a><span class="lineno"> 1157</span> <span class="comment"> * tables. Moreover, this function also creates the tree table</span></div> |
| <div class="line"><a name="l01158"></a><span class="lineno"> 1158</span> <span class="comment"> * specified by user.</span></div> |
| <div class="line"><a name="l01159"></a><span class="lineno"> 1159</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01160"></a><span class="lineno"> 1160</span> <span class="comment"> * @param result_tree_table_name The name of the tree specified by user. </span></div> |
| <div class="line"><a name="l01161"></a><span class="lineno"> 1161</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l01162"></a><span class="lineno"> 1162</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l01163"></a><span class="lineno"> 1163</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__create_tree_tables</div> |
| <div class="line"><a name="l01164"></a><span class="lineno"> 1164</span>  (</div> |
| <div class="line"><a name="l01165"></a><span class="lineno"> 1165</span>  result_tree_table_name TEXT</div> |
| <div class="line"><a name="l01166"></a><span class="lineno"> 1166</span>  ) </div> |
| <div class="line"><a name="l01167"></a><span class="lineno"> 1167</span> RETURNS <span class="keywordtype">void</span> AS $$ </div> |
| <div class="line"><a name="l01168"></a><span class="lineno"> 1168</span> BEGIN</div> |
| <div class="line"><a name="l01169"></a><span class="lineno"> 1169</span>  -- The table of node_size_aux records the size of each node. It is used</div> |
| <div class="line"><a name="l01170"></a><span class="lineno"> 1170</span>  -- for missing value handling.</div> |
| <div class="line"><a name="l01171"></a><span class="lineno"> 1171</span>  DROP TABLE IF EXISTS node_size_aux CASCADE;</div> |
| <div class="line"><a name="l01172"></a><span class="lineno"> 1172</span>  CREATE TEMP TABLE node_size_aux</div> |
| <div class="line"><a name="l01173"></a><span class="lineno"> 1173</span>  (</div> |
| <div class="line"><a name="l01174"></a><span class="lineno"> 1174</span>  tid INT,</div> |
| <div class="line"><a name="l01175"></a><span class="lineno"> 1175</span>  nid INT,</div> |
| <div class="line"><a name="l01176"></a><span class="lineno"> 1176</span>  node_size INT</div> |
| <div class="line"><a name="l01177"></a><span class="lineno"> 1177</span>  )m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (tid,nid)');</div> |
| <div class="line"><a name="l01178"></a><span class="lineno"> 1178</span>  </div> |
| <div class="line"><a name="l01179"></a><span class="lineno"> 1179</span>  -- The table below stores the decision tree information just constructed.</div> |
| <div class="line"><a name="l01180"></a><span class="lineno"> 1180</span>  -- Columns:</div> |
| <div class="line"><a name="l01181"></a><span class="lineno"> 1181</span>  -- <span class="keywordtype">id</span>: The ID of the node represented by this row. Tree </div> |
| <div class="line"><a name="l01182"></a><span class="lineno"> 1182</span>  -- node IDs are unique across all trees. The IDs of </div> |
| <div class="line"><a name="l01183"></a><span class="lineno"> 1183</span>  -- all children of a node is made to be continuous.</div> |
| <div class="line"><a name="l01184"></a><span class="lineno"> 1184</span>  -- tree_location: An array containing the encoded values of all the </div> |
| <div class="line"><a name="l01185"></a><span class="lineno"> 1185</span>  -- features on the path from the root node to the </div> |
| <div class="line"><a name="l01186"></a><span class="lineno"> 1186</span>  -- current node. For the root node, the location </div> |
| <div class="line"><a name="l01187"></a><span class="lineno"> 1187</span>  -- value is {0}.</div> |
| <div class="line"><a name="l01188"></a><span class="lineno"> 1188</span>  -- feature: The ID of the best split feature chosen <span class="keywordflow">for</span> the </div> |
| <div class="line"><a name="l01189"></a><span class="lineno"> 1189</span>  -- node represented by <span class="keyword">this</span> row.</div> |
| <div class="line"><a name="l01190"></a><span class="lineno"> 1190</span>  -- probability: If forced to make a call <span class="keywordflow">for</span> a dominant <span class="keyword">class </span></div> |
| <div class="line"><a name="l01191"></a><span class="lineno"> 1191</span>  -- at a given point this would be the confidence of the </div> |
| <div class="line"><a name="l01192"></a><span class="lineno"> 1192</span>  -- call (this is only an estimated value).</div> |
| <div class="line"><a name="l01193"></a><span class="lineno"> 1193</span>  -- ebp_coeff: The total errors used by error based pruning (ebp) </div> |
| <div class="line"><a name="l01194"></a><span class="lineno"> 1194</span>  -- based on the specified confidence level. RF does</div> |
| <div class="line"><a name="l01195"></a><span class="lineno"> 1195</span>  -- not do EBP therefore for RF nodes, this column always </div> |
| <div class="line"><a name="l01196"></a><span class="lineno"> 1196</span>  -- contains 1. </div> |
| <div class="line"><a name="l01197"></a><span class="lineno"> 1197</span>  -- max_class: If forced to make a call for a dominant class </div> |
| <div class="line"><a name="l01198"></a><span class="lineno"> 1198</span>  -- at a given point this is the selected class.</div> |
| <div class="line"><a name="l01199"></a><span class="lineno"> 1199</span>  -- scv: The splitting criteria value (scv) computed at this node.</div> |
| <div class="line"><a name="l01200"></a><span class="lineno"> 1200</span>  -- live: Specifies whether the node should be further split </div> |
| <div class="line"><a name="l01201"></a><span class="lineno"> 1201</span>  -- or not. A positive value indicates further split of</div> |
| <div class="line"><a name="l01202"></a><span class="lineno"> 1202</span>  -- the node represented by this row is needed.</div> |
| <div class="line"><a name="l01203"></a><span class="lineno"> 1203</span>  -- num_of_samples: The number of samples at this node.</div> |
| <div class="line"><a name="l01204"></a><span class="lineno"> 1204</span>  -- parent_id: Id of the parent branch.</div> |
| <div class="line"><a name="l01205"></a><span class="lineno"> 1205</span>  -- lmc_nid: Leftmost child (lmc) node id of the node represented</div> |
| <div class="line"><a name="l01206"></a><span class="lineno"> 1206</span>  -- by the current row.</div> |
| <div class="line"><a name="l01207"></a><span class="lineno"> 1207</span>  -- lmc_fval: The feature value which leads to the lmc node. </div> |
| <div class="line"><a name="l01208"></a><span class="lineno"> 1208</span>  -- An example of getting all the child nodes' ids </div> |
| <div class="line"><a name="l01209"></a><span class="lineno"> 1209</span>  -- and condition values</div> |
| <div class="line"><a name="l01210"></a><span class="lineno"> 1210</span>  -- 1. Get the right most node id</div> |
| <div class="line"><a name="l01211"></a><span class="lineno"> 1211</span>  -- SELECT DISTINCT ON(parent_id) id FROM tree_table </div> |
| <div class="line"><a name="l01212"></a><span class="lineno"> 1212</span>  -- WHERE parent_id = $pid ORDER BY parent_id, id desc</div> |
| <div class="line"><a name="l01213"></a><span class="lineno"> 1213</span>  -- INTO max_child_nid;</div> |
| <div class="line"><a name="l01214"></a><span class="lineno"> 1214</span>  -- 2. Get child nodes' ids and condition values by a </div> |
| <div class="line"><a name="l01215"></a><span class="lineno"> 1215</span>  -- while loop</div> |
| <div class="line"><a name="l01216"></a><span class="lineno"> 1216</span>  -- node_count = 1;</div> |
| <div class="line"><a name="l01217"></a><span class="lineno"> 1217</span>  -- WHILE (lmc_nid IS NOT NULL) AND </div> |
| <div class="line"><a name="l01218"></a><span class="lineno"> 1218</span>  -- (0 < node_count AND lmc_nid <= max_child_nid) LOOP</div> |
| <div class="line"><a name="l01219"></a><span class="lineno"> 1219</span>  -- ...</div> |
| <div class="line"><a name="l01220"></a><span class="lineno"> 1220</span>  -- lmc_nid = lmc_nid + 1;</div> |
| <div class="line"><a name="l01221"></a><span class="lineno"> 1221</span>  -- lmc_fval = lmc_fval + 1; </div> |
| <div class="line"><a name="l01222"></a><span class="lineno"> 1222</span>  -- SELECT COUNT(id) FROM tree_table </div> |
| <div class="line"><a name="l01223"></a><span class="lineno"> 1223</span>  -- WHERE id = $lmc_nid AND parent_id = $pid </div> |
| <div class="line"><a name="l01224"></a><span class="lineno"> 1224</span>  -- INTO node_count; </div> |
| <div class="line"><a name="l01225"></a><span class="lineno"> 1225</span>  -- END LOOP;</div> |
| <div class="line"><a name="l01226"></a><span class="lineno"> 1226</span>  -- is_cont: It specifies whether the selected feature is a </div> |
| <div class="line"><a name="l01227"></a><span class="lineno"> 1227</span>  -- continuous feature.</div> |
| <div class="line"><a name="l01228"></a><span class="lineno"> 1228</span>  -- split_value: For continuous feature, it specifies the split value.</div> |
| <div class="line"><a name="l01229"></a><span class="lineno"> 1229</span>  -- Otherwise, it is of no meaning and fixed to 0. </div> |
| <div class="line"><a name="l01230"></a><span class="lineno"> 1230</span>  -- tid: The id of a tree that this node belongs to.</div> |
| <div class="line"><a name="l01231"></a><span class="lineno"> 1231</span>  -- dp_ids: An array containing the IDs of the non-continuous </div> |
| <div class="line"><a name="l01232"></a><span class="lineno"> 1232</span>  -- features chosen by all ancestors nodes (starting </div> |
| <div class="line"><a name="l01233"></a><span class="lineno"> 1233</span>  -- from the root) for splitting.</div> |
| <div class="line"><a name="l01234"></a><span class="lineno"> 1234</span>  --</div> |
| <div class="line"><a name="l01235"></a><span class="lineno"> 1235</span>  -- The table below stores the final decision tree information.</div> |
| <div class="line"><a name="l01236"></a><span class="lineno"> 1236</span>  -- It is an the table specified by users. </div> |
| <div class="line"><a name="l01237"></a><span class="lineno"> 1237</span>  -- Please refer the table above for detailed column definition.</div> |
| <div class="line"><a name="l01238"></a><span class="lineno"> 1238</span>  EXECUTE 'DROP TABLE IF EXISTS '||result_tree_table_name||' CASCADE;';</div> |
| <div class="line"><a name="l01239"></a><span class="lineno"> 1239</span>  EXECUTE 'CREATE TABLE '||result_tree_table_name||'</div> |
| <div class="line"><a name="l01240"></a><span class="lineno"> 1240</span>  (</div> |
| <div class="line"><a name="l01241"></a><span class="lineno"> 1241</span>  id INT,</div> |
| <div class="line"><a name="l01242"></a><span class="lineno"> 1242</span>  tree_location INT[],</div> |
| <div class="line"><a name="l01243"></a><span class="lineno"> 1243</span>  feature INT,</div> |
| <div class="line"><a name="l01244"></a><span class="lineno"> 1244</span>  probability FLOAT,</div> |
| <div class="line"><a name="l01245"></a><span class="lineno"> 1245</span>  ebp_coeff FLOAT,</div> |
| <div class="line"><a name="l01246"></a><span class="lineno"> 1246</span>  max_class INTEGER,</div> |
| <div class="line"><a name="l01247"></a><span class="lineno"> 1247</span>  scv FLOAT,</div> |
| <div class="line"><a name="l01248"></a><span class="lineno"> 1248</span>  live INT,</div> |
| <div class="line"><a name="l01249"></a><span class="lineno"> 1249</span>  num_of_samples INT,</div> |
| <div class="line"><a name="l01250"></a><span class="lineno"> 1250</span>  parent_id INT,</div> |
| <div class="line"><a name="l01251"></a><span class="lineno"> 1251</span>  lmc_nid INT,</div> |
| <div class="line"><a name="l01252"></a><span class="lineno"> 1252</span>  lmc_fval INT,</div> |
| <div class="line"><a name="l01253"></a><span class="lineno"> 1253</span>  is_cont BOOLEAN,</div> |
| <div class="line"><a name="l01254"></a><span class="lineno"> 1254</span>  split_value FLOAT,</div> |
| <div class="line"><a name="l01255"></a><span class="lineno"> 1255</span>  tid INT,</div> |
| <div class="line"><a name="l01256"></a><span class="lineno"> 1256</span>  dp_ids INT[]</div> |
| <div class="line"><a name="l01257"></a><span class="lineno"> 1257</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (tid,id)');';</div> |
| <div class="line"><a name="l01258"></a><span class="lineno"> 1258</span>  </div> |
| <div class="line"><a name="l01259"></a><span class="lineno"> 1259</span>  -- The following table stored the auxiliary information for updating the </div> |
| <div class="line"><a name="l01260"></a><span class="lineno"> 1260</span>  -- association table, so that the updating operation only need to </div> |
| <div class="line"><a name="l01261"></a><span class="lineno"> 1261</span>  -- join the encoded table with association table once</div> |
| <div class="line"><a name="l01262"></a><span class="lineno"> 1262</span>  EXECUTE 'DROP TABLE IF EXISTS assoc_aux CASCADE';</div> |
| <div class="line"><a name="l01263"></a><span class="lineno"> 1263</span>  CREATE TEMP TABLE assoc_aux</div> |
| <div class="line"><a name="l01264"></a><span class="lineno"> 1264</span>  (</div> |
| <div class="line"><a name="l01265"></a><span class="lineno"> 1265</span>  nid INT, </div> |
| <div class="line"><a name="l01266"></a><span class="lineno"> 1266</span>  fid INT, </div> |
| <div class="line"><a name="l01267"></a><span class="lineno"> 1267</span>  lmc_id INT, </div> |
| <div class="line"><a name="l01268"></a><span class="lineno"> 1268</span>  svalue FLOAT, </div> |
| <div class="line"><a name="l01269"></a><span class="lineno"> 1269</span>  is_cont BOOL</div> |
| <div class="line"><a name="l01270"></a><span class="lineno"> 1270</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (nid)'); </div> |
| <div class="line"><a name="l01271"></a><span class="lineno"> 1271</span> </div> |
| <div class="line"><a name="l01272"></a><span class="lineno"> 1272</span>  EXECUTE 'DROP TABLE IF EXISTS tr_assoc_ping CASCADE';</div> |
| <div class="line"><a name="l01273"></a><span class="lineno"> 1273</span>  EXECUTE 'DROP TABLE IF EXISTS tr_assoc_pong CASCADE';</div> |
| <div class="line"><a name="l01274"></a><span class="lineno"> 1274</span>  EXECUTE 'DROP TABLE IF EXISTS sf_assoc CASCADE';</div> |
| <div class="line"><a name="l01275"></a><span class="lineno"> 1275</span>  </div> |
| <div class="line"><a name="l01276"></a><span class="lineno"> 1276</span> m4_changequote(`>>>', `<<<')</div> |
| <div class="line"><a name="l01277"></a><span class="lineno"> 1277</span> m4_ifdef(>>>__GREENPLUM_GE_4_2_1__<<<, >>></div> |
| <div class="line"><a name="l01278"></a><span class="lineno"> 1278</span>  CREATE TEMP TABLE tr_assoc_ping</div> |
| <div class="line"><a name="l01279"></a><span class="lineno"> 1279</span>  (</div> |
| <div class="line"><a name="l01280"></a><span class="lineno"> 1280</span>  id BIGINT ENCODING (compresstype=RLE_TYPE),</div> |
| <div class="line"><a name="l01281"></a><span class="lineno"> 1281</span>  nid INT ENCODING (compresstype=RLE_TYPE),</div> |
| <div class="line"><a name="l01282"></a><span class="lineno"> 1282</span>  tid INT ENCODING (compresstype=RLE_TYPE),</div> |
| <div class="line"><a name="l01283"></a><span class="lineno"> 1283</span>  weight INT ENCODING (compresstype=RLE_TYPE) </div> |
| <div class="line"><a name="l01284"></a><span class="lineno"> 1284</span>  )</div> |
| <div class="line"><a name="l01285"></a><span class="lineno"> 1285</span>  WITH(appendonly=true, orientation=column)</div> |
| <div class="line"><a name="l01286"></a><span class="lineno"> 1286</span>  DISTRIBUTED BY(id);</div> |
| <div class="line"><a name="l01287"></a><span class="lineno"> 1287</span> </div> |
| <div class="line"><a name="l01288"></a><span class="lineno"> 1288</span>  CREATE TEMP TABLE tr_assoc_pong</div> |
| <div class="line"><a name="l01289"></a><span class="lineno"> 1289</span>  (</div> |
| <div class="line"><a name="l01290"></a><span class="lineno"> 1290</span>  id BIGINT ENCODING (compresstype=RLE_TYPE),</div> |
| <div class="line"><a name="l01291"></a><span class="lineno"> 1291</span>  nid INT ENCODING (compresstype=RLE_TYPE),</div> |
| <div class="line"><a name="l01292"></a><span class="lineno"> 1292</span>  tid INT ENCODING (compresstype=RLE_TYPE),</div> |
| <div class="line"><a name="l01293"></a><span class="lineno"> 1293</span>  weight INT ENCODING (compresstype=RLE_TYPE)</div> |
| <div class="line"><a name="l01294"></a><span class="lineno"> 1294</span>  )</div> |
| <div class="line"><a name="l01295"></a><span class="lineno"> 1295</span>  WITH(appendonly=true, orientation=column)</div> |
| <div class="line"><a name="l01296"></a><span class="lineno"> 1296</span>  DISTRIBUTED BY(id);</div> |
| <div class="line"><a name="l01297"></a><span class="lineno"> 1297</span> </div> |
| <div class="line"><a name="l01298"></a><span class="lineno"> 1298</span>  CREATE TEMP TABLE sf_assoc</div> |
| <div class="line"><a name="l01299"></a><span class="lineno"> 1299</span>  (</div> |
| <div class="line"><a name="l01300"></a><span class="lineno"> 1300</span>  nid INT ENCODING (compresstype=RLE_TYPE),</div> |
| <div class="line"><a name="l01301"></a><span class="lineno"> 1301</span>  fid INT ENCODING (compresstype=RLE_TYPE)</div> |
| <div class="line"><a name="l01302"></a><span class="lineno"> 1302</span>  )</div> |
| <div class="line"><a name="l01303"></a><span class="lineno"> 1303</span>  WITH(appendonly=true, orientation=column)</div> |
| <div class="line"><a name="l01304"></a><span class="lineno"> 1304</span>  DISTRIBUTED BY(fid);</div> |
| <div class="line"><a name="l01305"></a><span class="lineno"> 1305</span> <<<, >>></div> |
| <div class="line"><a name="l01306"></a><span class="lineno"> 1306</span>  CREATE TEMP TABLE tr_assoc_ping</div> |
| <div class="line"><a name="l01307"></a><span class="lineno"> 1307</span>  (</div> |
| <div class="line"><a name="l01308"></a><span class="lineno"> 1308</span>  id BIGINT,</div> |
| <div class="line"><a name="l01309"></a><span class="lineno"> 1309</span>  nid INT,</div> |
| <div class="line"><a name="l01310"></a><span class="lineno"> 1310</span>  tid INT,</div> |
| <div class="line"><a name="l01311"></a><span class="lineno"> 1311</span>  weight INT</div> |
| <div class="line"><a name="l01312"></a><span class="lineno"> 1312</span>  )m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (id)');</div> |
| <div class="line"><a name="l01313"></a><span class="lineno"> 1313</span>  CREATE TEMP TABLE tr_assoc_pong</div> |
| <div class="line"><a name="l01314"></a><span class="lineno"> 1314</span>  (</div> |
| <div class="line"><a name="l01315"></a><span class="lineno"> 1315</span>  id BIGINT,</div> |
| <div class="line"><a name="l01316"></a><span class="lineno"> 1316</span>  nid INT,</div> |
| <div class="line"><a name="l01317"></a><span class="lineno"> 1317</span>  tid INT,</div> |
| <div class="line"><a name="l01318"></a><span class="lineno"> 1318</span>  weight INT</div> |
| <div class="line"><a name="l01319"></a><span class="lineno"> 1319</span>  )m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (id)');</div> |
| <div class="line"><a name="l01320"></a><span class="lineno"> 1320</span>  CREATE TEMP TABLE sf_assoc</div> |
| <div class="line"><a name="l01321"></a><span class="lineno"> 1321</span>  (</div> |
| <div class="line"><a name="l01322"></a><span class="lineno"> 1322</span>  nid INT,</div> |
| <div class="line"><a name="l01323"></a><span class="lineno"> 1323</span>  fid INT</div> |
| <div class="line"><a name="l01324"></a><span class="lineno"> 1324</span>  )m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (fid)'); </div> |
| <div class="line"><a name="l01325"></a><span class="lineno"> 1325</span> <<<)</div> |
| <div class="line"><a name="l01326"></a><span class="lineno"> 1326</span> m4_changequote(>>>`<<<, >>>'<<<) </div> |
| <div class="line"><a name="l01327"></a><span class="lineno"> 1327</span> END </div> |
| <div class="line"><a name="l01328"></a><span class="lineno"> 1328</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l01329"></a><span class="lineno"> 1329</span> </div> |
| <div class="line"><a name="l01330"></a><span class="lineno"> 1330</span> </div> |
| <div class="line"><a name="l01331"></a><span class="lineno"> 1331</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l01332"></a><span class="lineno"> 1332</span> <span class="comment"> * @brief Prune the trained tree with "Reduced Error Pruning" algorithm.</span></div> |
| <div class="line"><a name="l01333"></a><span class="lineno"> 1333</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01334"></a><span class="lineno"> 1334</span> <span class="comment"> * @param tree_table_name The name of the table containing the tree. </span></div> |
| <div class="line"><a name="l01335"></a><span class="lineno"> 1335</span> <span class="comment"> * @param validation_table The name of the table containing validation set. </span></div> |
| <div class="line"><a name="l01336"></a><span class="lineno"> 1336</span> <span class="comment"> * @param max_num_classes The count of different classes. </span></div> |
| <div class="line"><a name="l01337"></a><span class="lineno"> 1337</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l01338"></a><span class="lineno"> 1338</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l01339"></a><span class="lineno"> 1339</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__rep_prune_tree</div> |
| <div class="line"><a name="l01340"></a><span class="lineno"> 1340</span>  (</div> |
| <div class="line"><a name="l01341"></a><span class="lineno"> 1341</span>  tree_table_name TEXT, </div> |
| <div class="line"><a name="l01342"></a><span class="lineno"> 1342</span>  validation_table TEXT, </div> |
| <div class="line"><a name="l01343"></a><span class="lineno"> 1343</span>  max_num_classes INT</div> |
| <div class="line"><a name="l01344"></a><span class="lineno"> 1344</span>  ) </div> |
| <div class="line"><a name="l01345"></a><span class="lineno"> 1345</span> RETURNS void AS $$</div> |
| <div class="line"><a name="l01346"></a><span class="lineno"> 1346</span> DECLARE</div> |
| <div class="line"><a name="l01347"></a><span class="lineno"> 1347</span>  num_parent_ids INTEGER;</div> |
| <div class="line"><a name="l01348"></a><span class="lineno"> 1348</span>  cf_table_name TEXT;</div> |
| <div class="line"><a name="l01349"></a><span class="lineno"> 1349</span>  encoded_table_name TEXT;</div> |
| <div class="line"><a name="l01350"></a><span class="lineno"> 1350</span>  metatable_name TEXT;</div> |
| <div class="line"><a name="l01351"></a><span class="lineno"> 1351</span>  curstmt TEXT;</div> |
| <div class="line"><a name="l01352"></a><span class="lineno"> 1352</span>  id_col_name TEXT;</div> |
| <div class="line"><a name="l01353"></a><span class="lineno"> 1353</span>  class_col_name TEXT;</div> |
| <div class="line"><a name="l01354"></a><span class="lineno"> 1354</span>  classify_result TEXT;</div> |
| <div class="line"><a name="l01355"></a><span class="lineno"> 1355</span>  temp_text TEXT;</div> |
| <div class="line"><a name="l01356"></a><span class="lineno"> 1356</span>  n INT;</div> |
| <div class="line"><a name="l01357"></a><span class="lineno"> 1357</span>  table_names TEXT[];</div> |
| <div class="line"><a name="l01358"></a><span class="lineno"> 1358</span> BEGIN</div> |
| <div class="line"><a name="l01359"></a><span class="lineno"> 1359</span>  metatable_name = MADLIB_SCHEMA.__get_metatable_name(tree_table_name);</div> |
| <div class="line"><a name="l01360"></a><span class="lineno"> 1360</span>  id_col_name = MADLIB_SCHEMA.__get_id_column_name(metatable_name);</div> |
| <div class="line"><a name="l01361"></a><span class="lineno"> 1361</span>  class_col_name = MADLIB_SCHEMA.__get_class_column_name(metatable_name);</div> |
| <div class="line"><a name="l01362"></a><span class="lineno"> 1362</span>  </div> |
| <div class="line"><a name="l01363"></a><span class="lineno"> 1363</span>  -- the value of class column in validation table must in the KV table</div> |
| <div class="line"><a name="l01364"></a><span class="lineno"> 1364</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01365"></a><span class="lineno"> 1365</span>  (</div> |
| <div class="line"><a name="l01366"></a><span class="lineno"> 1366</span>  'SELECT COUNT(*) </div> |
| <div class="line"><a name="l01367"></a><span class="lineno"> 1367</span>  FROM %</div> |
| <div class="line"><a name="l01368"></a><span class="lineno"> 1368</span>  WHERE MADLIB_SCHEMA.__to_char(%) NOT IN</div> |
| <div class="line"><a name="l01369"></a><span class="lineno"> 1369</span>  (SELECT fval FROM % WHERE fval IS NOT NULL)',</div> |
| <div class="line"><a name="l01370"></a><span class="lineno"> 1370</span>  ARRAY[</div> |
| <div class="line"><a name="l01371"></a><span class="lineno"> 1371</span>  validation_table,</div> |
| <div class="line"><a name="l01372"></a><span class="lineno"> 1372</span>  class_col_name,</div> |
| <div class="line"><a name="l01373"></a><span class="lineno"> 1373</span>  MADLIB_SCHEMA.__get_classtable_name(metatable_name)</div> |
| <div class="line"><a name="l01374"></a><span class="lineno"> 1374</span>  ]</div> |
| <div class="line"><a name="l01375"></a><span class="lineno"> 1375</span>  )</div> |
| <div class="line"><a name="l01376"></a><span class="lineno"> 1376</span>  INTO curstmt;</div> |
| <div class="line"><a name="l01377"></a><span class="lineno"> 1377</span>  </div> |
| <div class="line"><a name="l01378"></a><span class="lineno"> 1378</span>  EXECUTE curstmt INTO n;</div> |
| <div class="line"><a name="l01379"></a><span class="lineno"> 1379</span>  </div> |
| <div class="line"><a name="l01380"></a><span class="lineno"> 1380</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l01381"></a><span class="lineno"> 1381</span>  (</div> |
| <div class="line"><a name="l01382"></a><span class="lineno"> 1382</span>  n = 0,</div> |
| <div class="line"><a name="l01383"></a><span class="lineno"> 1383</span>  'the value of class column in validation table must in </div> |
| <div class="line"><a name="l01384"></a><span class="lineno"> 1384</span>  training table'</div> |
| <div class="line"><a name="l01385"></a><span class="lineno"> 1385</span>  ); </div> |
| <div class="line"><a name="l01386"></a><span class="lineno"> 1386</span>  </div> |
| <div class="line"><a name="l01387"></a><span class="lineno"> 1387</span>  table_names = MADLIB_SCHEMA.__treemodel_classify_internal</div> |
| <div class="line"><a name="l01388"></a><span class="lineno"> 1388</span>  (</div> |
| <div class="line"><a name="l01389"></a><span class="lineno"> 1389</span>  validation_table, </div> |
| <div class="line"><a name="l01390"></a><span class="lineno"> 1390</span>  tree_table_name, </div> |
| <div class="line"><a name="l01391"></a><span class="lineno"> 1391</span>  0</div> |
| <div class="line"><a name="l01392"></a><span class="lineno"> 1392</span>  );</div> |
| <div class="line"><a name="l01393"></a><span class="lineno"> 1393</span>  </div> |
| <div class="line"><a name="l01394"></a><span class="lineno"> 1394</span>  encoded_table_name = table_names[1];</div> |
| <div class="line"><a name="l01395"></a><span class="lineno"> 1395</span>  classify_result = table_names[2];</div> |
| <div class="line"><a name="l01396"></a><span class="lineno"> 1396</span>  cf_table_name = classify_result;</div> |
| <div class="line"><a name="l01397"></a><span class="lineno"> 1397</span> </div> |
| <div class="line"><a name="l01398"></a><span class="lineno"> 1398</span>  -- after encoding in classification, class_col_name is fixed to class</div> |
| <div class="line"><a name="l01399"></a><span class="lineno"> 1399</span>  class_col_name = 'class';</div> |
| <div class="line"><a name="l01400"></a><span class="lineno"> 1400</span>  </div> |
| <div class="line"><a name="l01401"></a><span class="lineno"> 1401</span> m4_changequote(`>>>', `<<<')</div> |
| <div class="line"><a name="l01402"></a><span class="lineno"> 1402</span> m4_ifdef(>>>__GREENPLUM_PRE_4_1__<<<, >>></div> |
| <div class="line"><a name="l01403"></a><span class="lineno"> 1403</span>  EXECUTE 'DROP TABLE IF EXISTS tree_rep_pong CASCADE';</div> |
| <div class="line"><a name="l01404"></a><span class="lineno"> 1404</span>  EXECUTE 'CREATE TEMP TABLE tree_rep_pong AS SELECT * FROM ' || </div> |
| <div class="line"><a name="l01405"></a><span class="lineno"> 1405</span>  classify_result || </div> |
| <div class="line"><a name="l01406"></a><span class="lineno"> 1406</span>  ' LIMIT 0 m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (id)')';</div> |
| <div class="line"><a name="l01407"></a><span class="lineno"> 1407</span> <<<)</div> |
| <div class="line"><a name="l01408"></a><span class="lineno"> 1408</span> m4_changequote(>>>`<<<, >>>'<<<)</div> |
| <div class="line"><a name="l01409"></a><span class="lineno"> 1409</span>  </div> |
| <div class="line"><a name="l01410"></a><span class="lineno"> 1410</span>  LOOP</div> |
| <div class="line"><a name="l01411"></a><span class="lineno"> 1411</span>  DROP TABLE IF EXISTS selected_parent_ids_rep;</div> |
| <div class="line"><a name="l01412"></a><span class="lineno"> 1412</span>  CREATE TEMP TABLE selected_parent_ids_rep</div> |
| <div class="line"><a name="l01413"></a><span class="lineno"> 1413</span>  (</div> |
| <div class="line"><a name="l01414"></a><span class="lineno"> 1414</span>  parent_id BIGINT,</div> |
| <div class="line"><a name="l01415"></a><span class="lineno"> 1415</span>  max_class INT</div> |
| <div class="line"><a name="l01416"></a><span class="lineno"> 1416</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (parent_id)');</div> |
| <div class="line"><a name="l01417"></a><span class="lineno"> 1417</span>  </div> |
| <div class="line"><a name="l01418"></a><span class="lineno"> 1418</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01419"></a><span class="lineno"> 1419</span>  (</div> |
| <div class="line"><a name="l01420"></a><span class="lineno"> 1420</span>  'INSERT INTO selected_parent_ids_rep </div> |
| <div class="line"><a name="l01421"></a><span class="lineno"> 1421</span>  SELECT parent_id, t.g[1] as max_class </div> |
| <div class="line"><a name="l01422"></a><span class="lineno"> 1422</span>  FROM </div> |
| <div class="line"><a name="l01423"></a><span class="lineno"> 1423</span>  (</div> |
| <div class="line"><a name="l01424"></a><span class="lineno"> 1424</span>  SELECT parent_id, </div> |
| <div class="line"><a name="l01425"></a><span class="lineno"> 1425</span>  MADLIB_SCHEMA.__rep_aggr_class_count</div> |
| <div class="line"><a name="l01426"></a><span class="lineno"> 1426</span>  (</div> |
| <div class="line"><a name="l01427"></a><span class="lineno"> 1427</span>  c.class, </div> |
| <div class="line"><a name="l01428"></a><span class="lineno"> 1428</span>  s.%, </div> |
| <div class="line"><a name="l01429"></a><span class="lineno"> 1429</span>  % </div> |
| <div class="line"><a name="l01430"></a><span class="lineno"> 1430</span>  ) AS g </div> |
| <div class="line"><a name="l01431"></a><span class="lineno"> 1431</span>  FROM % c, % s </div> |
| <div class="line"><a name="l01432"></a><span class="lineno"> 1432</span>  WHERE c.id=s.% </div> |
| <div class="line"><a name="l01433"></a><span class="lineno"> 1433</span>  GROUP BY parent_id</div> |
| <div class="line"><a name="l01434"></a><span class="lineno"> 1434</span>  ) t </div> |
| <div class="line"><a name="l01435"></a><span class="lineno"> 1435</span>  WHERE t.g[2] >= 0 AND </div> |
| <div class="line"><a name="l01436"></a><span class="lineno"> 1436</span>  t.parent_id IN </div> |
| <div class="line"><a name="l01437"></a><span class="lineno"> 1437</span>  (</div> |
| <div class="line"><a name="l01438"></a><span class="lineno"> 1438</span>  Select parent_id FROM % </div> |
| <div class="line"><a name="l01439"></a><span class="lineno"> 1439</span>  WHERE parent_id NOT IN</div> |
| <div class="line"><a name="l01440"></a><span class="lineno"> 1440</span>  (</div> |
| <div class="line"><a name="l01441"></a><span class="lineno"> 1441</span>  Select parent_id </div> |
| <div class="line"><a name="l01442"></a><span class="lineno"> 1442</span>  FROM % </div> |
| <div class="line"><a name="l01443"></a><span class="lineno"> 1443</span>  WHERE lmc_nid IS NOT NULL</div> |
| <div class="line"><a name="l01444"></a><span class="lineno"> 1444</span>  ) and id <> 1</div> |
| <div class="line"><a name="l01445"></a><span class="lineno"> 1445</span>  );',</div> |
| <div class="line"><a name="l01446"></a><span class="lineno"> 1446</span>  ARRAY[</div> |
| <div class="line"><a name="l01447"></a><span class="lineno"> 1447</span>  class_col_name,</div> |
| <div class="line"><a name="l01448"></a><span class="lineno"> 1448</span>  MADLIB_SCHEMA.__to_char(max_num_classes),</div> |
| <div class="line"><a name="l01449"></a><span class="lineno"> 1449</span>  classify_result,</div> |
| <div class="line"><a name="l01450"></a><span class="lineno"> 1450</span>  encoded_table_name,</div> |
| <div class="line"><a name="l01451"></a><span class="lineno"> 1451</span>  id_col_name,</div> |
| <div class="line"><a name="l01452"></a><span class="lineno"> 1452</span>  tree_table_name,</div> |
| <div class="line"><a name="l01453"></a><span class="lineno"> 1453</span>  tree_table_name</div> |
| <div class="line"><a name="l01454"></a><span class="lineno"> 1454</span>  ]</div> |
| <div class="line"><a name="l01455"></a><span class="lineno"> 1455</span>  )</div> |
| <div class="line"><a name="l01456"></a><span class="lineno"> 1456</span>  INTO curstmt;</div> |
| <div class="line"><a name="l01457"></a><span class="lineno"> 1457</span>  </div> |
| <div class="line"><a name="l01458"></a><span class="lineno"> 1458</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l01459"></a><span class="lineno"> 1459</span>  </div> |
| <div class="line"><a name="l01460"></a><span class="lineno"> 1460</span>  EXECUTE 'SELECT parent_id FROM selected_parent_ids_rep limit 1;' </div> |
| <div class="line"><a name="l01461"></a><span class="lineno"> 1461</span>  INTO num_parent_ids;</div> |
| <div class="line"><a name="l01462"></a><span class="lineno"> 1462</span>  IF (num_parent_ids IS NULL) THEN</div> |
| <div class="line"><a name="l01463"></a><span class="lineno"> 1463</span>  EXIT;</div> |
| <div class="line"><a name="l01464"></a><span class="lineno"> 1464</span>  END IF;</div> |
| <div class="line"><a name="l01465"></a><span class="lineno"> 1465</span> </div> |
| <div class="line"><a name="l01466"></a><span class="lineno"> 1466</span> m4_changequote(`>>>', `<<<')</div> |
| <div class="line"><a name="l01467"></a><span class="lineno"> 1467</span> m4_ifdef(`__GREENPLUM_PRE_4_1__', >>></div> |
| <div class="line"><a name="l01468"></a><span class="lineno"> 1468</span>  -- for some databases, update operation can't distribute data across segments</div> |
| <div class="line"><a name="l01469"></a><span class="lineno"> 1469</span>  -- we use two tables to update the data</div> |
| <div class="line"><a name="l01470"></a><span class="lineno"> 1470</span>  IF (classify_result = 'tree_rep_pong') THEN</div> |
| <div class="line"><a name="l01471"></a><span class="lineno"> 1471</span>  temp_text = cf_table_name;</div> |
| <div class="line"><a name="l01472"></a><span class="lineno"> 1472</span>  ELSE</div> |
| <div class="line"><a name="l01473"></a><span class="lineno"> 1473</span>  temp_text = 'tree_rep_pong';</div> |
| <div class="line"><a name="l01474"></a><span class="lineno"> 1474</span>  END IF;</div> |
| <div class="line"><a name="l01475"></a><span class="lineno"> 1475</span>  </div> |
| <div class="line"><a name="l01476"></a><span class="lineno"> 1476</span>  EXECUTE 'TRUNCATE ' || temp_text;</div> |
| <div class="line"><a name="l01477"></a><span class="lineno"> 1477</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01478"></a><span class="lineno"> 1478</span>  (</div> |
| <div class="line"><a name="l01479"></a><span class="lineno"> 1479</span>  'INSERT INTO %(id, class, parent_id, leaf_id)</div> |
| <div class="line"><a name="l01480"></a><span class="lineno"> 1480</span>  SELECT m.id, t.max_class, t.parent_id, t.id</div> |
| <div class="line"><a name="l01481"></a><span class="lineno"> 1481</span>  FROM % m, % t</div> |
| <div class="line"><a name="l01482"></a><span class="lineno"> 1482</span>  WHERE t.id IN (SELECT parent_id FROM selected_parent_ids_rep) AND</div> |
| <div class="line"><a name="l01483"></a><span class="lineno"> 1483</span>  m.parent_id = t.id',</div> |
| <div class="line"><a name="l01484"></a><span class="lineno"> 1484</span>  ARRAY[</div> |
| <div class="line"><a name="l01485"></a><span class="lineno"> 1485</span>  temp_text,</div> |
| <div class="line"><a name="l01486"></a><span class="lineno"> 1486</span>  classify_result,</div> |
| <div class="line"><a name="l01487"></a><span class="lineno"> 1487</span>  tree_table_name</div> |
| <div class="line"><a name="l01488"></a><span class="lineno"> 1488</span>  ]</div> |
| <div class="line"><a name="l01489"></a><span class="lineno"> 1489</span>  )</div> |
| <div class="line"><a name="l01490"></a><span class="lineno"> 1490</span>  INTO curstmt;</div> |
| <div class="line"><a name="l01491"></a><span class="lineno"> 1491</span>  </div> |
| <div class="line"><a name="l01492"></a><span class="lineno"> 1492</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l01493"></a><span class="lineno"> 1493</span>  </div> |
| <div class="line"><a name="l01494"></a><span class="lineno"> 1494</span>  classify_result = temp_text;</div> |
| <div class="line"><a name="l01495"></a><span class="lineno"> 1495</span> <<<, >>></div> |
| <div class="line"><a name="l01496"></a><span class="lineno"> 1496</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01497"></a><span class="lineno"> 1497</span>  (</div> |
| <div class="line"><a name="l01498"></a><span class="lineno"> 1498</span>  'UPDATE % m set class = t.max_class, </div> |
| <div class="line"><a name="l01499"></a><span class="lineno"> 1499</span>  parent_id = t.parent_id,leaf_id = t.id </div> |
| <div class="line"><a name="l01500"></a><span class="lineno"> 1500</span>  FROM % t</div> |
| <div class="line"><a name="l01501"></a><span class="lineno"> 1501</span>  WHERE t.id IN (SELECT parent_id FROM selected_parent_ids_rep) AND</div> |
| <div class="line"><a name="l01502"></a><span class="lineno"> 1502</span>  m.parent_id=t.id',</div> |
| <div class="line"><a name="l01503"></a><span class="lineno"> 1503</span>  classify_result,</div> |
| <div class="line"><a name="l01504"></a><span class="lineno"> 1504</span>  tree_table_name</div> |
| <div class="line"><a name="l01505"></a><span class="lineno"> 1505</span>  )</div> |
| <div class="line"><a name="l01506"></a><span class="lineno"> 1506</span>  INTO curstmt;</div> |
| <div class="line"><a name="l01507"></a><span class="lineno"> 1507</span>  EXECUTE curstmt; </div> |
| <div class="line"><a name="l01508"></a><span class="lineno"> 1508</span> <<<)</div> |
| <div class="line"><a name="l01509"></a><span class="lineno"> 1509</span> m4_changequote(>>>`<<<, >>>'<<<)</div> |
| <div class="line"><a name="l01510"></a><span class="lineno"> 1510</span>  </div> |
| <div class="line"><a name="l01511"></a><span class="lineno"> 1511</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01512"></a><span class="lineno"> 1512</span>  (</div> |
| <div class="line"><a name="l01513"></a><span class="lineno"> 1513</span>  'DELETE FROM % WHERE parent_id IN </div> |
| <div class="line"><a name="l01514"></a><span class="lineno"> 1514</span>  (SELECT parent_id FROM selected_parent_ids_rep)',</div> |
| <div class="line"><a name="l01515"></a><span class="lineno"> 1515</span>  tree_table_name</div> |
| <div class="line"><a name="l01516"></a><span class="lineno"> 1516</span>  )</div> |
| <div class="line"><a name="l01517"></a><span class="lineno"> 1517</span>  INTO curstmt;</div> |
| <div class="line"><a name="l01518"></a><span class="lineno"> 1518</span>  </div> |
| <div class="line"><a name="l01519"></a><span class="lineno"> 1519</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l01520"></a><span class="lineno"> 1520</span> </div> |
| <div class="line"><a name="l01521"></a><span class="lineno"> 1521</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01522"></a><span class="lineno"> 1522</span>  (</div> |
| <div class="line"><a name="l01523"></a><span class="lineno"> 1523</span>  'UPDATE % t1 SET lmc_nid = NULL, </div> |
| <div class="line"><a name="l01524"></a><span class="lineno"> 1524</span>  lmc_fval = NULL, max_class = t2.max_class </div> |
| <div class="line"><a name="l01525"></a><span class="lineno"> 1525</span>  FROM selected_parent_ids_rep t2</div> |
| <div class="line"><a name="l01526"></a><span class="lineno"> 1526</span>  WHERE t1.id = t2.parent_id;',</div> |
| <div class="line"><a name="l01527"></a><span class="lineno"> 1527</span>  tree_table_name</div> |
| <div class="line"><a name="l01528"></a><span class="lineno"> 1528</span>  )</div> |
| <div class="line"><a name="l01529"></a><span class="lineno"> 1529</span>  INTO curstmt;</div> |
| <div class="line"><a name="l01530"></a><span class="lineno"> 1530</span>  </div> |
| <div class="line"><a name="l01531"></a><span class="lineno"> 1531</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l01532"></a><span class="lineno"> 1532</span>  </div> |
| <div class="line"><a name="l01533"></a><span class="lineno"> 1533</span>  END LOOP;</div> |
| <div class="line"><a name="l01534"></a><span class="lineno"> 1534</span>  </div> |
| <div class="line"><a name="l01535"></a><span class="lineno"> 1535</span>  EXECUTE 'DROP TABLE IF EXISTS ' || encoded_table_name || ' CASCADE;';</div> |
| <div class="line"><a name="l01536"></a><span class="lineno"> 1536</span> END</div> |
| <div class="line"><a name="l01537"></a><span class="lineno"> 1537</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l01538"></a><span class="lineno"> 1538</span> </div> |
| <div class="line"><a name="l01539"></a><span class="lineno"> 1539</span> </div> |
| <div class="line"><a name="l01540"></a><span class="lineno"> 1540</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l01541"></a><span class="lineno"> 1541</span> <span class="comment"> * @brief Calculates the total errors used by Error Based Pruning (EBP).</span></div> |
| <div class="line"><a name="l01542"></a><span class="lineno"> 1542</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01543"></a><span class="lineno"> 1543</span> <span class="comment"> * @param total The number of total samples represented by the node </span></div> |
| <div class="line"><a name="l01544"></a><span class="lineno"> 1544</span> <span class="comment"> * being processed. </span></div> |
| <div class="line"><a name="l01545"></a><span class="lineno"> 1545</span> <span class="comment"> * @param prob The probability to mis-classify samples represented by the </span></div> |
| <div class="line"><a name="l01546"></a><span class="lineno"> 1546</span> <span class="comment"> * child nodes if they are pruned with EBP. </span></div> |
| <div class="line"><a name="l01547"></a><span class="lineno"> 1547</span> <span class="comment"> * @param confidence_level A certainty factor to calculate the confidence limits</span></div> |
| <div class="line"><a name="l01548"></a><span class="lineno"> 1548</span> <span class="comment"> * for the probability of error using the binomial theorem. </span></div> |
| <div class="line"><a name="l01549"></a><span class="lineno"> 1549</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l01550"></a><span class="lineno"> 1550</span> <span class="comment"> * @return The computed total error.</span></div> |
| <div class="line"><a name="l01551"></a><span class="lineno"> 1551</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01552"></a><span class="lineno"> 1552</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l01553"></a><span class="lineno"> 1553</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__ebp_calc_errors</div> |
| <div class="line"><a name="l01554"></a><span class="lineno"> 1554</span>  (</div> |
| <div class="line"><a name="l01555"></a><span class="lineno"> 1555</span>  total FLOAT8,</div> |
| <div class="line"><a name="l01556"></a><span class="lineno"> 1556</span>  prob FLOAT8,</div> |
| <div class="line"><a name="l01557"></a><span class="lineno"> 1557</span>  confidence_level FLOAT8</div> |
| <div class="line"><a name="l01558"></a><span class="lineno"> 1558</span>  ) RETURNS FLOAT8 </div> |
| <div class="line"><a name="l01559"></a><span class="lineno"> 1559</span> AS 'MODULE_PATHNAME', 'dt_ebp_calc_errors'</div> |
| <div class="line"><a name="l01560"></a><span class="lineno"> 1560</span> LANGUAGE C STRICT IMMUTABLE;</div> |
| <div class="line"><a name="l01561"></a><span class="lineno"> 1561</span> </div> |
| <div class="line"><a name="l01562"></a><span class="lineno"> 1562</span> </div> |
| <div class="line"><a name="l01563"></a><span class="lineno"> 1563</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l01564"></a><span class="lineno"> 1564</span> <span class="comment"> * @brief Prune the trained tree with "Error-based Pruning" algorithm.</span></div> |
| <div class="line"><a name="l01565"></a><span class="lineno"> 1565</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01566"></a><span class="lineno"> 1566</span> <span class="comment"> * @param tree_table_name The name of the table containing the tree. </span></div> |
| <div class="line"><a name="l01567"></a><span class="lineno"> 1567</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l01568"></a><span class="lineno"> 1568</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l01569"></a><span class="lineno"> 1569</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__ebp_prune_tree</div> |
| <div class="line"><a name="l01570"></a><span class="lineno"> 1570</span>  (</div> |
| <div class="line"><a name="l01571"></a><span class="lineno"> 1571</span>  tree_table_name TEXT</div> |
| <div class="line"><a name="l01572"></a><span class="lineno"> 1572</span>  ) </div> |
| <div class="line"><a name="l01573"></a><span class="lineno"> 1573</span> RETURNS void AS $$</div> |
| <div class="line"><a name="l01574"></a><span class="lineno"> 1574</span> DECLARE</div> |
| <div class="line"><a name="l01575"></a><span class="lineno"> 1575</span>  num_parent_ids INTEGER;</div> |
| <div class="line"><a name="l01576"></a><span class="lineno"> 1576</span>  curstmt TEXT;</div> |
| <div class="line"><a name="l01577"></a><span class="lineno"> 1577</span> BEGIN</div> |
| <div class="line"><a name="l01578"></a><span class="lineno"> 1578</span>  LOOP</div> |
| <div class="line"><a name="l01579"></a><span class="lineno"> 1579</span>  DROP TABLE IF EXISTS selected_parent_ids_ebp;</div> |
| <div class="line"><a name="l01580"></a><span class="lineno"> 1580</span>  CREATE TEMP TABLE selected_parent_ids_ebp(parent_id BIGINT) </div> |
| <div class="line"><a name="l01581"></a><span class="lineno"> 1581</span>  m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY(parent_id)');</div> |
| <div class="line"><a name="l01582"></a><span class="lineno"> 1582</span>  </div> |
| <div class="line"><a name="l01583"></a><span class="lineno"> 1583</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01584"></a><span class="lineno"> 1584</span>  (</div> |
| <div class="line"><a name="l01585"></a><span class="lineno"> 1585</span>  'INSERT INTO selected_parent_ids_ebp </div> |
| <div class="line"><a name="l01586"></a><span class="lineno"> 1586</span>  SELECT s.parent_id as parent_id </div> |
| <div class="line"><a name="l01587"></a><span class="lineno"> 1587</span>  FROM </div> |
| <div class="line"><a name="l01588"></a><span class="lineno"> 1588</span>  (</div> |
| <div class="line"><a name="l01589"></a><span class="lineno"> 1589</span>  Select parent_id, sum(ebp_coeff) as ebp_coeff </div> |
| <div class="line"><a name="l01590"></a><span class="lineno"> 1590</span>  FROM </div> |
| <div class="line"><a name="l01591"></a><span class="lineno"> 1591</span>  (</div> |
| <div class="line"><a name="l01592"></a><span class="lineno"> 1592</span>  Select parent_id, ebp_coeff </div> |
| <div class="line"><a name="l01593"></a><span class="lineno"> 1593</span>  FROM % </div> |
| <div class="line"><a name="l01594"></a><span class="lineno"> 1594</span>  WHERE parent_id NOT IN </div> |
| <div class="line"><a name="l01595"></a><span class="lineno"> 1595</span>  (</div> |
| <div class="line"><a name="l01596"></a><span class="lineno"> 1596</span>  Select parent_id FROM % WHERE lmc_nid IS NOT NULL</div> |
| <div class="line"><a name="l01597"></a><span class="lineno"> 1597</span>  ) and id <> 1</div> |
| <div class="line"><a name="l01598"></a><span class="lineno"> 1598</span>  ) m </div> |
| <div class="line"><a name="l01599"></a><span class="lineno"> 1599</span>  GROUP BY m.parent_id</div> |
| <div class="line"><a name="l01600"></a><span class="lineno"> 1600</span>  ) s </div> |
| <div class="line"><a name="l01601"></a><span class="lineno"> 1601</span>  LEFT JOIN % p </div> |
| <div class="line"><a name="l01602"></a><span class="lineno"> 1602</span>  ON p.id = s.parent_id </div> |
| <div class="line"><a name="l01603"></a><span class="lineno"> 1603</span>  WHERE p.ebp_coeff < s.ebp_coeff;',</div> |
| <div class="line"><a name="l01604"></a><span class="lineno"> 1604</span>  tree_table_name,</div> |
| <div class="line"><a name="l01605"></a><span class="lineno"> 1605</span>  tree_table_name,</div> |
| <div class="line"><a name="l01606"></a><span class="lineno"> 1606</span>  tree_table_name</div> |
| <div class="line"><a name="l01607"></a><span class="lineno"> 1607</span>  )</div> |
| <div class="line"><a name="l01608"></a><span class="lineno"> 1608</span>  INTO curstmt;</div> |
| <div class="line"><a name="l01609"></a><span class="lineno"> 1609</span>  </div> |
| <div class="line"><a name="l01610"></a><span class="lineno"> 1610</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l01611"></a><span class="lineno"> 1611</span>  </div> |
| <div class="line"><a name="l01612"></a><span class="lineno"> 1612</span>  EXECUTE 'SELECT parent_id FROM selected_parent_ids_ebp LIMIT 1;' </div> |
| <div class="line"><a name="l01613"></a><span class="lineno"> 1613</span>  INTO num_parent_ids;</div> |
| <div class="line"><a name="l01614"></a><span class="lineno"> 1614</span> </div> |
| <div class="line"><a name="l01615"></a><span class="lineno"> 1615</span>  IF (num_parent_ids IS NULL) THEN</div> |
| <div class="line"><a name="l01616"></a><span class="lineno"> 1616</span>  EXIT;</div> |
| <div class="line"><a name="l01617"></a><span class="lineno"> 1617</span>  END IF;</div> |
| <div class="line"><a name="l01618"></a><span class="lineno"> 1618</span>  </div> |
| <div class="line"><a name="l01619"></a><span class="lineno"> 1619</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01620"></a><span class="lineno"> 1620</span>  (</div> |
| <div class="line"><a name="l01621"></a><span class="lineno"> 1621</span>  'DELETE FROM % </div> |
| <div class="line"><a name="l01622"></a><span class="lineno"> 1622</span>  WHERE parent_id IN </div> |
| <div class="line"><a name="l01623"></a><span class="lineno"> 1623</span>  (SELECT parent_id FROM selected_parent_ids_ebp)',</div> |
| <div class="line"><a name="l01624"></a><span class="lineno"> 1624</span>  tree_table_name</div> |
| <div class="line"><a name="l01625"></a><span class="lineno"> 1625</span>  )</div> |
| <div class="line"><a name="l01626"></a><span class="lineno"> 1626</span>  INTO curstmt;</div> |
| <div class="line"><a name="l01627"></a><span class="lineno"> 1627</span>  </div> |
| <div class="line"><a name="l01628"></a><span class="lineno"> 1628</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l01629"></a><span class="lineno"> 1629</span>  </div> |
| <div class="line"><a name="l01630"></a><span class="lineno"> 1630</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01631"></a><span class="lineno"> 1631</span>  (</div> |
| <div class="line"><a name="l01632"></a><span class="lineno"> 1632</span>  'UPDATE % </div> |
| <div class="line"><a name="l01633"></a><span class="lineno"> 1633</span>  SET lmc_nid = NULL, lmc_fval = NULL </div> |
| <div class="line"><a name="l01634"></a><span class="lineno"> 1634</span>  WHERE id IN </div> |
| <div class="line"><a name="l01635"></a><span class="lineno"> 1635</span>  (SELECT parent_id FROM selected_parent_ids_ebp)',</div> |
| <div class="line"><a name="l01636"></a><span class="lineno"> 1636</span>  tree_table_name</div> |
| <div class="line"><a name="l01637"></a><span class="lineno"> 1637</span>  )</div> |
| <div class="line"><a name="l01638"></a><span class="lineno"> 1638</span>  INTO curstmt;</div> |
| <div class="line"><a name="l01639"></a><span class="lineno"> 1639</span>  </div> |
| <div class="line"><a name="l01640"></a><span class="lineno"> 1640</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l01641"></a><span class="lineno"> 1641</span>  </div> |
| <div class="line"><a name="l01642"></a><span class="lineno"> 1642</span>  END LOOP;</div> |
| <div class="line"><a name="l01643"></a><span class="lineno"> 1643</span> END</div> |
| <div class="line"><a name="l01644"></a><span class="lineno"> 1644</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l01645"></a><span class="lineno"> 1645</span> </div> |
| <div class="line"><a name="l01646"></a><span class="lineno"> 1646</span> </div> |
| <div class="line"><a name="l01647"></a><span class="lineno"> 1647</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l01648"></a><span class="lineno"> 1648</span> <span class="comment"> * @brief Generate the final trained tree.</span></div> |
| <div class="line"><a name="l01649"></a><span class="lineno"> 1649</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01650"></a><span class="lineno"> 1650</span> <span class="comment"> * @param result_tree_table_name The name of the table containing the tree.</span></div> |
| <div class="line"><a name="l01651"></a><span class="lineno"> 1651</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l01652"></a><span class="lineno"> 1652</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l01653"></a><span class="lineno"> 1653</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__generate_final_tree</div> |
| <div class="line"><a name="l01654"></a><span class="lineno"> 1654</span>  (</div> |
| <div class="line"><a name="l01655"></a><span class="lineno"> 1655</span>  result_tree_table_name TEXT</div> |
| <div class="line"><a name="l01656"></a><span class="lineno"> 1656</span>  ) </div> |
| <div class="line"><a name="l01657"></a><span class="lineno"> 1657</span> RETURNS void AS $$</div> |
| <div class="line"><a name="l01658"></a><span class="lineno"> 1658</span> DECLARE</div> |
| <div class="line"><a name="l01659"></a><span class="lineno"> 1659</span>  tree_size INTEGER;</div> |
| <div class="line"><a name="l01660"></a><span class="lineno"> 1660</span>  curstmt TEXT;</div> |
| <div class="line"><a name="l01661"></a><span class="lineno"> 1661</span>  num_redundant_nodes INTEGER;</div> |
| <div class="line"><a name="l01662"></a><span class="lineno"> 1662</span> BEGIN</div> |
| <div class="line"><a name="l01663"></a><span class="lineno"> 1663</span> </div> |
| <div class="line"><a name="l01664"></a><span class="lineno"> 1664</span>  EXECUTE ' DELETE FROM ' || result_tree_table_name || </div> |
| <div class="line"><a name="l01665"></a><span class="lineno"> 1665</span>  ' WHERE COALESCE(num_of_samples,0) = 0';</div> |
| <div class="line"><a name="l01666"></a><span class="lineno"> 1666</span>  </div> |
| <div class="line"><a name="l01667"></a><span class="lineno"> 1667</span>  -- for each node, find the left most child node id and the feature value,</div> |
| <div class="line"><a name="l01668"></a><span class="lineno"> 1668</span>  -- and update the node's lmc_nid and lmc_fval column </div> |
| <div class="line"><a name="l01669"></a><span class="lineno"> 1669</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01670"></a><span class="lineno"> 1670</span>  (</div> |
| <div class="line"><a name="l01671"></a><span class="lineno"> 1671</span>  'UPDATE % k</div> |
| <div class="line"><a name="l01672"></a><span class="lineno"> 1672</span>  SET lmc_nid = g.lmc_nid, lmc_fval = g.lmc_fval</div> |
| <div class="line"><a name="l01673"></a><span class="lineno"> 1673</span>  FROM </div> |
| <div class="line"><a name="l01674"></a><span class="lineno"> 1674</span>  (</div> |
| <div class="line"><a name="l01675"></a><span class="lineno"> 1675</span>  SELECT parent_id,</div> |
| <div class="line"><a name="l01676"></a><span class="lineno"> 1676</span>  min(id) as lmc_nid,</div> |
| <div class="line"><a name="l01677"></a><span class="lineno"> 1677</span>  min(tree_location[array_upper(tree_location,1)]) </div> |
| <div class="line"><a name="l01678"></a><span class="lineno"> 1678</span>  as lmc_fval</div> |
| <div class="line"><a name="l01679"></a><span class="lineno"> 1679</span>  FROM %</div> |
| <div class="line"><a name="l01680"></a><span class="lineno"> 1680</span>  GROUP BY parent_id</div> |
| <div class="line"><a name="l01681"></a><span class="lineno"> 1681</span>  ) g</div> |
| <div class="line"><a name="l01682"></a><span class="lineno"> 1682</span>  WHERE k.id = g.parent_id',</div> |
| <div class="line"><a name="l01683"></a><span class="lineno"> 1683</span>  ARRAY[</div> |
| <div class="line"><a name="l01684"></a><span class="lineno"> 1684</span>  result_tree_table_name,</div> |
| <div class="line"><a name="l01685"></a><span class="lineno"> 1685</span>  result_tree_table_name</div> |
| <div class="line"><a name="l01686"></a><span class="lineno"> 1686</span>  ]</div> |
| <div class="line"><a name="l01687"></a><span class="lineno"> 1687</span>  )</div> |
| <div class="line"><a name="l01688"></a><span class="lineno"> 1688</span>  INTO curstmt; </div> |
| <div class="line"><a name="l01689"></a><span class="lineno"> 1689</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l01690"></a><span class="lineno"> 1690</span> </div> |
| <div class="line"><a name="l01691"></a><span class="lineno"> 1691</span>  <span class="comment">/*</span></div> |
| <div class="line"><a name="l01692"></a><span class="lineno"> 1692</span> <span class="comment"> * For a certain node, if all of its children are leaf nodes and have the </span></div> |
| <div class="line"><a name="l01693"></a><span class="lineno"> 1693</span> <span class="comment"> * same class label, we can safely remove its children. After removal, we</span></div> |
| <div class="line"><a name="l01694"></a><span class="lineno"> 1694</span> <span class="comment"> * should apply the same operation to the new leaf nodes until no nodes </span></div> |
| <div class="line"><a name="l01695"></a><span class="lineno"> 1695</span> <span class="comment"> * meet this criterion.</span></div> |
| <div class="line"><a name="l01696"></a><span class="lineno"> 1696</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l01697"></a><span class="lineno"> 1697</span>  LOOP</div> |
| <div class="line"><a name="l01698"></a><span class="lineno"> 1698</span>  EXECUTE 'DROP TABLE IF EXISTS trim_tree_aux_table CASCADE';</div> |
| <div class="line"><a name="l01699"></a><span class="lineno"> 1699</span>  -- Find nodes whose children should be removed.</div> |
| <div class="line"><a name="l01700"></a><span class="lineno"> 1700</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01701"></a><span class="lineno"> 1701</span>  (</div> |
| <div class="line"><a name="l01702"></a><span class="lineno"> 1702</span>  'CREATE TEMP TABLE trim_tree_aux_table AS</div> |
| <div class="line"><a name="l01703"></a><span class="lineno"> 1703</span>  SELECT parent_id FROM</div> |
| <div class="line"><a name="l01704"></a><span class="lineno"> 1704</span>  (</div> |
| <div class="line"><a name="l01705"></a><span class="lineno"> 1705</span>  SELECT parent_id, count(distinct max_class) as class_count </div> |
| <div class="line"><a name="l01706"></a><span class="lineno"> 1706</span>  FROM %</div> |
| <div class="line"><a name="l01707"></a><span class="lineno"> 1707</span>  WHERE parent_id IN</div> |
| <div class="line"><a name="l01708"></a><span class="lineno"> 1708</span>  (</div> |
| <div class="line"><a name="l01709"></a><span class="lineno"> 1709</span>  SELECT parent_id FROM % </div> |
| <div class="line"><a name="l01710"></a><span class="lineno"> 1710</span>  WHERE parent_id NOT IN</div> |
| <div class="line"><a name="l01711"></a><span class="lineno"> 1711</span>  (</div> |
| <div class="line"><a name="l01712"></a><span class="lineno"> 1712</span>  SELECT parent_id </div> |
| <div class="line"><a name="l01713"></a><span class="lineno"> 1713</span>  FROM % </div> |
| <div class="line"><a name="l01714"></a><span class="lineno"> 1714</span>  WHERE lmc_nid IS NOT NULL</div> |
| <div class="line"><a name="l01715"></a><span class="lineno"> 1715</span>  ) and parent_id <> 0</div> |
| <div class="line"><a name="l01716"></a><span class="lineno"> 1716</span>  ) </div> |
| <div class="line"><a name="l01717"></a><span class="lineno"> 1717</span>  GROUP BY parent_id</div> |
| <div class="line"><a name="l01718"></a><span class="lineno"> 1718</span>  ) l</div> |
| <div class="line"><a name="l01719"></a><span class="lineno"> 1719</span>  where l.class_count=1</div> |
| <div class="line"><a name="l01720"></a><span class="lineno"> 1720</span>  m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (parent_id)')',</div> |
| <div class="line"><a name="l01721"></a><span class="lineno"> 1721</span>  ARRAY[</div> |
| <div class="line"><a name="l01722"></a><span class="lineno"> 1722</span>  result_tree_table_name,</div> |
| <div class="line"><a name="l01723"></a><span class="lineno"> 1723</span>  result_tree_table_name,</div> |
| <div class="line"><a name="l01724"></a><span class="lineno"> 1724</span>  result_tree_table_name</div> |
| <div class="line"><a name="l01725"></a><span class="lineno"> 1725</span>  ]</div> |
| <div class="line"><a name="l01726"></a><span class="lineno"> 1726</span>  );</div> |
| <div class="line"><a name="l01727"></a><span class="lineno"> 1727</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l01728"></a><span class="lineno"> 1728</span> </div> |
| <div class="line"><a name="l01729"></a><span class="lineno"> 1729</span>  EXECUTE 'SELECT count(*) FROM trim_tree_aux_table' </div> |
| <div class="line"><a name="l01730"></a><span class="lineno"> 1730</span>  INTO num_redundant_nodes;</div> |
| <div class="line"><a name="l01731"></a><span class="lineno"> 1731</span> </div> |
| <div class="line"><a name="l01732"></a><span class="lineno"> 1732</span>  IF (num_redundant_nodes <= 0) THEN</div> |
| <div class="line"><a name="l01733"></a><span class="lineno"> 1733</span>  EXIT;</div> |
| <div class="line"><a name="l01734"></a><span class="lineno"> 1734</span>  END IF;</div> |
| <div class="line"><a name="l01735"></a><span class="lineno"> 1735</span>  </div> |
| <div class="line"><a name="l01736"></a><span class="lineno"> 1736</span>  -- Delete the found redundant nodes.</div> |
| <div class="line"><a name="l01737"></a><span class="lineno"> 1737</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01738"></a><span class="lineno"> 1738</span>  (</div> |
| <div class="line"><a name="l01739"></a><span class="lineno"> 1739</span>  '</div> |
| <div class="line"><a name="l01740"></a><span class="lineno"> 1740</span>  DELETE FROM % t</div> |
| <div class="line"><a name="l01741"></a><span class="lineno"> 1741</span>  WHERE t.parent_id IN</div> |
| <div class="line"><a name="l01742"></a><span class="lineno"> 1742</span>  (SELECT parent_id FROM trim_tree_aux_table)',</div> |
| <div class="line"><a name="l01743"></a><span class="lineno"> 1743</span>  ARRAY[</div> |
| <div class="line"><a name="l01744"></a><span class="lineno"> 1744</span>  result_tree_table_name</div> |
| <div class="line"><a name="l01745"></a><span class="lineno"> 1745</span>  ]</div> |
| <div class="line"><a name="l01746"></a><span class="lineno"> 1746</span>  );</div> |
| <div class="line"><a name="l01747"></a><span class="lineno"> 1747</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l01748"></a><span class="lineno"> 1748</span> </div> |
| <div class="line"><a name="l01749"></a><span class="lineno"> 1749</span>  -- Set the nodes, whose children are removed, to be leaf nodes.</div> |
| <div class="line"><a name="l01750"></a><span class="lineno"> 1750</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01751"></a><span class="lineno"> 1751</span>  (</div> |
| <div class="line"><a name="l01752"></a><span class="lineno"> 1752</span>  'UPDATE % k</div> |
| <div class="line"><a name="l01753"></a><span class="lineno"> 1753</span>  SET lmc_nid = NULL, lmc_fval = NULL</div> |
| <div class="line"><a name="l01754"></a><span class="lineno"> 1754</span>  FROM </div> |
| <div class="line"><a name="l01755"></a><span class="lineno"> 1755</span>  (</div> |
| <div class="line"><a name="l01756"></a><span class="lineno"> 1756</span>  SELECT parent_id FROM trim_tree_aux_table</div> |
| <div class="line"><a name="l01757"></a><span class="lineno"> 1757</span>  ) g</div> |
| <div class="line"><a name="l01758"></a><span class="lineno"> 1758</span>  WHERE k.id = g.parent_id',</div> |
| <div class="line"><a name="l01759"></a><span class="lineno"> 1759</span>  ARRAY[</div> |
| <div class="line"><a name="l01760"></a><span class="lineno"> 1760</span>  result_tree_table_name</div> |
| <div class="line"><a name="l01761"></a><span class="lineno"> 1761</span>  ]</div> |
| <div class="line"><a name="l01762"></a><span class="lineno"> 1762</span>  );</div> |
| <div class="line"><a name="l01763"></a><span class="lineno"> 1763</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l01764"></a><span class="lineno"> 1764</span>  END LOOP;</div> |
| <div class="line"><a name="l01765"></a><span class="lineno"> 1765</span> END</div> |
| <div class="line"><a name="l01766"></a><span class="lineno"> 1766</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l01767"></a><span class="lineno"> 1767</span> </div> |
| <div class="line"><a name="l01768"></a><span class="lineno"> 1768</span> </div> |
| <div class="line"><a name="l01769"></a><span class="lineno"> 1769</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l01770"></a><span class="lineno"> 1770</span> <span class="comment"> * The UDT for the training result.</span></div> |
| <div class="line"><a name="l01771"></a><span class="lineno"> 1771</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01772"></a><span class="lineno"> 1772</span> <span class="comment"> * num_of_samples It means how many records there exists in the </span></div> |
| <div class="line"><a name="l01773"></a><span class="lineno"> 1773</span> <span class="comment"> * training set. </span></div> |
| <div class="line"><a name="l01774"></a><span class="lineno"> 1774</span> <span class="comment"> * features_per_node The number of features chosen for each tree.</span></div> |
| <div class="line"><a name="l01775"></a><span class="lineno"> 1775</span> <span class="comment"> * num_tree_nodes The number of tree nodes.</span></div> |
| <div class="line"><a name="l01776"></a><span class="lineno"> 1776</span> <span class="comment"> * max_tree_depth The max tree depth.</span></div> |
| <div class="line"><a name="l01777"></a><span class="lineno"> 1777</span> <span class="comment"> * calc_acc_time Total time of calculating acc.</span></div> |
| <div class="line"><a name="l01778"></a><span class="lineno"> 1778</span> <span class="comment"> * calc_pre_time Time of preprocessing when calculating acc.</span></div> |
| <div class="line"><a name="l01779"></a><span class="lineno"> 1779</span> <span class="comment"> * update_time Total time of updating operation after found</span></div> |
| <div class="line"><a name="l01780"></a><span class="lineno"> 1780</span> <span class="comment"> * the best time. </span></div> |
| <div class="line"><a name="l01781"></a><span class="lineno"> 1781</span> <span class="comment"> * update_best Time of updating the best splits' information.</span></div> |
| <div class="line"><a name="l01782"></a><span class="lineno"> 1782</span> <span class="comment"> * update_child Time of generating the child nodes.</span></div> |
| <div class="line"><a name="l01783"></a><span class="lineno"> 1783</span> <span class="comment"> * update_nid Time of updating the assigned node IDs.</span></div> |
| <div class="line"><a name="l01784"></a><span class="lineno"> 1784</span> <span class="comment"> * scv_acs_time Time of calculating the best splits. </span></div> |
| <div class="line"><a name="l01785"></a><span class="lineno"> 1785</span> <span class="comment"> * prune_time Time of tree pruning.</span></div> |
| <div class="line"><a name="l01786"></a><span class="lineno"> 1786</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01787"></a><span class="lineno"> 1787</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l01788"></a><span class="lineno"> 1788</span> DROP TYPE IF EXISTS MADLIB_SCHEMA.__train_result;</div> |
| <div class="line"><a name="l01789"></a><span class="lineno"> 1789</span> CREATE TYPE MADLIB_SCHEMA.__train_result AS </div> |
| <div class="line"><a name="l01790"></a><span class="lineno"> 1790</span> ( </div> |
| <div class="line"><a name="l01791"></a><span class="lineno"> 1791</span>  num_of_samples BIGINT, </div> |
| <div class="line"><a name="l01792"></a><span class="lineno"> 1792</span>  features_per_node INT,</div> |
| <div class="line"><a name="l01793"></a><span class="lineno"> 1793</span>  num_tree_nodes INT,</div> |
| <div class="line"><a name="l01794"></a><span class="lineno"> 1794</span>  max_tree_depth INT,</div> |
| <div class="line"><a name="l01795"></a><span class="lineno"> 1795</span>  calc_acc_time INTERVAL,</div> |
| <div class="line"><a name="l01796"></a><span class="lineno"> 1796</span>  calc_pre_time INTERVAL,</div> |
| <div class="line"><a name="l01797"></a><span class="lineno"> 1797</span>  update_time INTERVAL,</div> |
| <div class="line"><a name="l01798"></a><span class="lineno"> 1798</span>  update_best INTERVAL,</div> |
| <div class="line"><a name="l01799"></a><span class="lineno"> 1799</span>  update_child INTERVAL,</div> |
| <div class="line"><a name="l01800"></a><span class="lineno"> 1800</span>  update_nid INTERVAL,</div> |
| <div class="line"><a name="l01801"></a><span class="lineno"> 1801</span>  scv_acs_time INTERVAL,</div> |
| <div class="line"><a name="l01802"></a><span class="lineno"> 1802</span>  prune_time INTERVAL</div> |
| <div class="line"><a name="l01803"></a><span class="lineno"> 1803</span> );</div> |
| <div class="line"><a name="l01804"></a><span class="lineno"> 1804</span> </div> |
| <div class="line"><a name="l01805"></a><span class="lineno"> 1805</span> </div> |
| <div class="line"><a name="l01806"></a><span class="lineno"> 1806</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l01807"></a><span class="lineno"> 1807</span> <span class="comment"> * @brief The function samples a set of integer values between low and high.</span></div> |
| <div class="line"><a name="l01808"></a><span class="lineno"> 1808</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01809"></a><span class="lineno"> 1809</span> <span class="comment"> * @param num_of_samples The number of records to be sampled.</span></div> |
| <div class="line"><a name="l01810"></a><span class="lineno"> 1810</span> <span class="comment"> * @param low The low limit of sampled values.</span></div> |
| <div class="line"><a name="l01811"></a><span class="lineno"> 1811</span> <span class="comment"> * @param high The high limit of sampled values.</span></div> |
| <div class="line"><a name="l01812"></a><span class="lineno"> 1812</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01813"></a><span class="lineno"> 1813</span> <span class="comment"> * @return A set of integer values sampled randomly between [low, high].</span></div> |
| <div class="line"><a name="l01814"></a><span class="lineno"> 1814</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01815"></a><span class="lineno"> 1815</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l01816"></a><span class="lineno"> 1816</span> DROP FUNCTION IF EXISTS MADLIB_SCHEMA.__sample_within_range</div> |
| <div class="line"><a name="l01817"></a><span class="lineno"> 1817</span>  (</div> |
| <div class="line"><a name="l01818"></a><span class="lineno"> 1818</span>  BIGINT,</div> |
| <div class="line"><a name="l01819"></a><span class="lineno"> 1819</span>  BIGINT,</div> |
| <div class="line"><a name="l01820"></a><span class="lineno"> 1820</span>  BIGINT</div> |
| <div class="line"><a name="l01821"></a><span class="lineno"> 1821</span>  )CASCADE;</div> |
| <div class="line"><a name="l01822"></a><span class="lineno"> 1822</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__sample_within_range</div> |
| <div class="line"><a name="l01823"></a><span class="lineno"> 1823</span>  (</div> |
| <div class="line"><a name="l01824"></a><span class="lineno"> 1824</span>  num_of_samples BIGINT,</div> |
| <div class="line"><a name="l01825"></a><span class="lineno"> 1825</span>  low BIGINT,</div> |
| <div class="line"><a name="l01826"></a><span class="lineno"> 1826</span>  high BIGINT</div> |
| <div class="line"><a name="l01827"></a><span class="lineno"> 1827</span>  ) </div> |
| <div class="line"><a name="l01828"></a><span class="lineno"> 1828</span> RETURNS SETOF BIGINT </div> |
| <div class="line"><a name="l01829"></a><span class="lineno"> 1829</span> AS 'MODULE_PATHNAME', 'dt_sample_within_range'</div> |
| <div class="line"><a name="l01830"></a><span class="lineno"> 1830</span> LANGUAGE C STRICT VOLATILE;</div> |
| <div class="line"><a name="l01831"></a><span class="lineno"> 1831</span> </div> |
| <div class="line"><a name="l01832"></a><span class="lineno"> 1832</span> </div> |
| <div class="line"><a name="l01833"></a><span class="lineno"> 1833</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l01834"></a><span class="lineno"> 1834</span> <span class="comment"> * @brief The function samples with replacement from source table and store</span></div> |
| <div class="line"><a name="l01835"></a><span class="lineno"> 1835</span> <span class="comment"> * the results to target table.</span></div> |
| <div class="line"><a name="l01836"></a><span class="lineno"> 1836</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l01837"></a><span class="lineno"> 1837</span> <span class="comment"> * In this function, we firstly calculate how many samples should be</span></div> |
| <div class="line"><a name="l01838"></a><span class="lineno"> 1838</span> <span class="comment"> * generated in each segment. Then, we let those segments sample with</span></div> |
| <div class="line"><a name="l01839"></a><span class="lineno"> 1839</span> <span class="comment"> * replacement between the maximum ID and minimum ID of the source table </span></div> |
| <div class="line"><a name="l01840"></a><span class="lineno"> 1840</span> <span class="comment"> * in parallel and assign samples to different trees. </span></div> |
| <div class="line"><a name="l01841"></a><span class="lineno"> 1841</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01842"></a><span class="lineno"> 1842</span> <span class="comment"> * If there are gaps in the ID column of the source table, we sample</span></div> |
| <div class="line"><a name="l01843"></a><span class="lineno"> 1843</span> <span class="comment"> * extra records in proportion to the number of gaps. At last, we remove</span></div> |
| <div class="line"><a name="l01844"></a><span class="lineno"> 1844</span> <span class="comment"> * these invalid samples with an inner join operation with the source</span></div> |
| <div class="line"><a name="l01845"></a><span class="lineno"> 1845</span> <span class="comment"> * table. Since we target big data, this strategy works quite well.</span></div> |
| <div class="line"><a name="l01846"></a><span class="lineno"> 1846</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01847"></a><span class="lineno"> 1847</span> <span class="comment"> * @param num_of_tree The number of trees to be trained.</span></div> |
| <div class="line"><a name="l01848"></a><span class="lineno"> 1848</span> <span class="comment"> * @param size_per_tree The number of records to be sampled for each tree.</span></div> |
| <div class="line"><a name="l01849"></a><span class="lineno"> 1849</span> <span class="comment"> * @param src_table The name of the table to be sampled from.</span></div> |
| <div class="line"><a name="l01850"></a><span class="lineno"> 1850</span> <span class="comment"> * @param target_table The name of the table used to store the results.</span></div> |
| <div class="line"><a name="l01851"></a><span class="lineno"> 1851</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01852"></a><span class="lineno"> 1852</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l01853"></a><span class="lineno"> 1853</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__sample_with_replacement</div> |
| <div class="line"><a name="l01854"></a><span class="lineno"> 1854</span>  ( </div> |
| <div class="line"><a name="l01855"></a><span class="lineno"> 1855</span>  num_of_tree INT,</div> |
| <div class="line"><a name="l01856"></a><span class="lineno"> 1856</span>  size_per_tree INT, </div> |
| <div class="line"><a name="l01857"></a><span class="lineno"> 1857</span>  src_table TEXT,</div> |
| <div class="line"><a name="l01858"></a><span class="lineno"> 1858</span>  target_table TEXT</div> |
| <div class="line"><a name="l01859"></a><span class="lineno"> 1859</span>  ) </div> |
| <div class="line"><a name="l01860"></a><span class="lineno"> 1860</span> RETURNS VOID AS $$</div> |
| <div class="line"><a name="l01861"></a><span class="lineno"> 1861</span> DECLARE</div> |
| <div class="line"><a name="l01862"></a><span class="lineno"> 1862</span>  segment_num INT;</div> |
| <div class="line"><a name="l01863"></a><span class="lineno"> 1863</span>  sample_per_seg INT;</div> |
| <div class="line"><a name="l01864"></a><span class="lineno"> 1864</span>  sample_ratio FLOAT8;</div> |
| <div class="line"><a name="l01865"></a><span class="lineno"> 1865</span>  record_num FLOAT8;</div> |
| <div class="line"><a name="l01866"></a><span class="lineno"> 1866</span>  min_id INT;</div> |
| <div class="line"><a name="l01867"></a><span class="lineno"> 1867</span>  max_id INT;</div> |
| <div class="line"><a name="l01868"></a><span class="lineno"> 1868</span>  range FLOAT8;</div> |
| <div class="line"><a name="l01869"></a><span class="lineno"> 1869</span>  stmt TEXT;</div> |
| <div class="line"><a name="l01870"></a><span class="lineno"> 1870</span> BEGIN</div> |
| <div class="line"><a name="l01871"></a><span class="lineno"> 1871</span> </div> |
| <div class="line"><a name="l01872"></a><span class="lineno"> 1872</span> m4_changequote(`>>>', `<<<')</div> |
| <div class="line"><a name="l01873"></a><span class="lineno"> 1873</span> m4_ifdef(>>>__GREENPLUM__<<<, >>></div> |
| <div class="line"><a name="l01874"></a><span class="lineno"> 1874</span>  -- get the segment number</div> |
| <div class="line"><a name="l01875"></a><span class="lineno"> 1875</span>  SELECT COUNT(distinct content) FROM gp_segment_configuration </div> |
| <div class="line"><a name="l01876"></a><span class="lineno"> 1876</span>  WHERE content<>-1 INTO segment_num;</div> |
| <div class="line"><a name="l01877"></a><span class="lineno"> 1877</span> <<<, >>></div> |
| <div class="line"><a name="l01878"></a><span class="lineno"> 1878</span>  -- fix the segment number to 1 for PG</div> |
| <div class="line"><a name="l01879"></a><span class="lineno"> 1879</span>  segment_num = 1;</div> |
| <div class="line"><a name="l01880"></a><span class="lineno"> 1880</span> <<<)</div> |
| <div class="line"><a name="l01881"></a><span class="lineno"> 1881</span> m4_changequote(>>>`<<<, >>>'<<<)</div> |
| <div class="line"><a name="l01882"></a><span class="lineno"> 1882</span> </div> |
| <div class="line"><a name="l01883"></a><span class="lineno"> 1883</span> </div> |
| <div class="line"><a name="l01884"></a><span class="lineno"> 1884</span>  DROP TABLE IF EXISTS auxiliary_segment_table;</div> |
| <div class="line"><a name="l01885"></a><span class="lineno"> 1885</span>  CREATE TEMP TABLE auxiliary_segment_table</div> |
| <div class="line"><a name="l01886"></a><span class="lineno"> 1886</span>  (</div> |
| <div class="line"><a name="l01887"></a><span class="lineno"> 1887</span>  segment_id INT </div> |
| <div class="line"><a name="l01888"></a><span class="lineno"> 1888</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY(segment_id)');</div> |
| <div class="line"><a name="l01889"></a><span class="lineno"> 1889</span> </div> |
| <div class="line"><a name="l01890"></a><span class="lineno"> 1890</span>  -- Insert segment_num of records distributed by segment id</div> |
| <div class="line"><a name="l01891"></a><span class="lineno"> 1891</span>  EXECUTE 'INSERT INTO auxiliary_segment_table </div> |
| <div class="line"><a name="l01892"></a><span class="lineno"> 1892</span>  SELECT generate_series(1,'||segment_num||');';</div> |
| <div class="line"><a name="l01893"></a><span class="lineno"> 1893</span> </div> |
| <div class="line"><a name="l01894"></a><span class="lineno"> 1894</span>  EXECUTE 'SELECT max(id),min(id), count(id) as record_num </div> |
| <div class="line"><a name="l01895"></a><span class="lineno"> 1895</span>  FROM '||src_table||';' INTO max_id,min_id,record_num;</div> |
| <div class="line"><a name="l01896"></a><span class="lineno"> 1896</span>  range=max_id-min_id+1;</div> |
| <div class="line"><a name="l01897"></a><span class="lineno"> 1897</span> </div> |
| <div class="line"><a name="l01898"></a><span class="lineno"> 1898</span>  -- compute the sample ratio</div> |
| <div class="line"><a name="l01899"></a><span class="lineno"> 1899</span>  sample_ratio= range/record_num;</div> |
| <div class="line"><a name="l01900"></a><span class="lineno"> 1900</span> </div> |
| <div class="line"><a name="l01901"></a><span class="lineno"> 1901</span>  -- compute how many records should be sampled by each segment</div> |
| <div class="line"><a name="l01902"></a><span class="lineno"> 1902</span>  sample_per_seg=((sample_ratio*num_of_tree*size_per_tree)/segment_num)::INT;</div> |
| <div class="line"><a name="l01903"></a><span class="lineno"> 1903</span>  </div> |
| <div class="line"><a name="l01904"></a><span class="lineno"> 1904</span>  -- add the weight field</div> |
| <div class="line"><a name="l01905"></a><span class="lineno"> 1905</span> </div> |
| <div class="line"><a name="l01906"></a><span class="lineno"> 1906</span>  IF (range > record_num) THEN</div> |
| <div class="line"><a name="l01907"></a><span class="lineno"> 1907</span>  -- remove those invalid samples with join operation</div> |
| <div class="line"><a name="l01908"></a><span class="lineno"> 1908</span>  stmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01909"></a><span class="lineno"> 1909</span>  (</div> |
| <div class="line"><a name="l01910"></a><span class="lineno"> 1910</span>  'INSERT INTO %(id, tid, nid, weight)</div> |
| <div class="line"><a name="l01911"></a><span class="lineno"> 1911</span>  SELECT record_id, </div> |
| <div class="line"><a name="l01912"></a><span class="lineno"> 1912</span>  tid AS tid, </div> |
| <div class="line"><a name="l01913"></a><span class="lineno"> 1913</span>  tid AS nid, </div> |
| <div class="line"><a name="l01914"></a><span class="lineno"> 1914</span>  count(*) AS weight</div> |
| <div class="line"><a name="l01915"></a><span class="lineno"> 1915</span>  FROM</div> |
| <div class="line"><a name="l01916"></a><span class="lineno"> 1916</span>  (</div> |
| <div class="line"><a name="l01917"></a><span class="lineno"> 1917</span>  SELECT MADLIB_SCHEMA.__sample_within_range(%, %, %) AS record_id,</div> |
| <div class="line"><a name="l01918"></a><span class="lineno"> 1918</span>  MADLIB_SCHEMA.__sample_within_range(%, 1, %) AS tid </div> |
| <div class="line"><a name="l01919"></a><span class="lineno"> 1919</span>  FROM auxiliary_segment_table</div> |
| <div class="line"><a name="l01920"></a><span class="lineno"> 1920</span>  ) t, </div> |
| <div class="line"><a name="l01921"></a><span class="lineno"> 1921</span>  % k</div> |
| <div class="line"><a name="l01922"></a><span class="lineno"> 1922</span>  WHERE t.record_id=k.id</div> |
| <div class="line"><a name="l01923"></a><span class="lineno"> 1923</span>  GROUP BY record_id, tid, nid',</div> |
| <div class="line"><a name="l01924"></a><span class="lineno"> 1924</span>  ARRAY[</div> |
| <div class="line"><a name="l01925"></a><span class="lineno"> 1925</span>  target_table,</div> |
| <div class="line"><a name="l01926"></a><span class="lineno"> 1926</span>  sample_per_seg::TEXT,</div> |
| <div class="line"><a name="l01927"></a><span class="lineno"> 1927</span>  min_id::TEXT,</div> |
| <div class="line"><a name="l01928"></a><span class="lineno"> 1928</span>  max_id::TEXT,</div> |
| <div class="line"><a name="l01929"></a><span class="lineno"> 1929</span>  sample_per_seg::TEXT,</div> |
| <div class="line"><a name="l01930"></a><span class="lineno"> 1930</span>  num_of_tree::TEXT,</div> |
| <div class="line"><a name="l01931"></a><span class="lineno"> 1931</span>  src_table</div> |
| <div class="line"><a name="l01932"></a><span class="lineno"> 1932</span>  ]</div> |
| <div class="line"><a name="l01933"></a><span class="lineno"> 1933</span>  );</div> |
| <div class="line"><a name="l01934"></a><span class="lineno"> 1934</span>  ELSE</div> |
| <div class="line"><a name="l01935"></a><span class="lineno"> 1935</span>  stmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l01936"></a><span class="lineno"> 1936</span>  (</div> |
| <div class="line"><a name="l01937"></a><span class="lineno"> 1937</span>  'INSERT INTO %(id, tid, nid, weight)</div> |
| <div class="line"><a name="l01938"></a><span class="lineno"> 1938</span>  SELECT record_id, </div> |
| <div class="line"><a name="l01939"></a><span class="lineno"> 1939</span>  tid AS tid, </div> |
| <div class="line"><a name="l01940"></a><span class="lineno"> 1940</span>  tid AS nid, </div> |
| <div class="line"><a name="l01941"></a><span class="lineno"> 1941</span>  count(*) AS weight</div> |
| <div class="line"><a name="l01942"></a><span class="lineno"> 1942</span>  FROM</div> |
| <div class="line"><a name="l01943"></a><span class="lineno"> 1943</span>  (</div> |
| <div class="line"><a name="l01944"></a><span class="lineno"> 1944</span>  SELECT MADLIB_SCHEMA.__sample_within_range(%, %, %) AS record_id,</div> |
| <div class="line"><a name="l01945"></a><span class="lineno"> 1945</span>  MADLIB_SCHEMA.__sample_within_range(%, 1, %) AS tid </div> |
| <div class="line"><a name="l01946"></a><span class="lineno"> 1946</span>  FROM auxiliary_segment_table</div> |
| <div class="line"><a name="l01947"></a><span class="lineno"> 1947</span>  ) t </div> |
| <div class="line"><a name="l01948"></a><span class="lineno"> 1948</span>  GROUP BY record_id, tid, nid',</div> |
| <div class="line"><a name="l01949"></a><span class="lineno"> 1949</span>  ARRAY[</div> |
| <div class="line"><a name="l01950"></a><span class="lineno"> 1950</span>  target_table,</div> |
| <div class="line"><a name="l01951"></a><span class="lineno"> 1951</span>  sample_per_seg::TEXT,</div> |
| <div class="line"><a name="l01952"></a><span class="lineno"> 1952</span>  min_id::TEXT,</div> |
| <div class="line"><a name="l01953"></a><span class="lineno"> 1953</span>  max_id::TEXT,</div> |
| <div class="line"><a name="l01954"></a><span class="lineno"> 1954</span>  sample_per_seg::TEXT,</div> |
| <div class="line"><a name="l01955"></a><span class="lineno"> 1955</span>  num_of_tree::TEXT</div> |
| <div class="line"><a name="l01956"></a><span class="lineno"> 1956</span>  ]</div> |
| <div class="line"><a name="l01957"></a><span class="lineno"> 1957</span>  );</div> |
| <div class="line"><a name="l01958"></a><span class="lineno"> 1958</span>  END IF;</div> |
| <div class="line"><a name="l01959"></a><span class="lineno"> 1959</span> </div> |
| <div class="line"><a name="l01960"></a><span class="lineno"> 1960</span>  EXECUTE stmt;</div> |
| <div class="line"><a name="l01961"></a><span class="lineno"> 1961</span> END</div> |
| <div class="line"><a name="l01962"></a><span class="lineno"> 1962</span> $$ LANGUAGE PLPGSQL VOLATILE;</div> |
| <div class="line"><a name="l01963"></a><span class="lineno"> 1963</span> </div> |
| <div class="line"><a name="l01964"></a><span class="lineno"> 1964</span> </div> |
| <div class="line"><a name="l01965"></a><span class="lineno"> 1965</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l01966"></a><span class="lineno"> 1966</span> <span class="comment"> * @brief This function trains a decision tree or random forest.</span></div> |
| <div class="line"><a name="l01967"></a><span class="lineno"> 1967</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01968"></a><span class="lineno"> 1968</span> <span class="comment"> * @param split_criterion This parameter specifies which split criterion </span></div> |
| <div class="line"><a name="l01969"></a><span class="lineno"> 1969</span> <span class="comment"> * should be used for tree construction and </span></div> |
| <div class="line"><a name="l01970"></a><span class="lineno"> 1970</span> <span class="comment"> * pruning. The valid values are infogain, </span></div> |
| <div class="line"><a name="l01971"></a><span class="lineno"> 1971</span> <span class="comment"> * gainratio, and gini.</span></div> |
| <div class="line"><a name="l01972"></a><span class="lineno"> 1972</span> <span class="comment"> * @param num_trees Total number of trees to be trained. </span></div> |
| <div class="line"><a name="l01973"></a><span class="lineno"> 1973</span> <span class="comment"> * @param features_per_node Total number of features used to compute split </span></div> |
| <div class="line"><a name="l01974"></a><span class="lineno"> 1974</span> <span class="comment"> * gain for each node. </span></div> |
| <div class="line"><a name="l01975"></a><span class="lineno"> 1975</span> <span class="comment"> * @param training_table_name The name of the table/view with the source data. </span></div> |
| <div class="line"><a name="l01976"></a><span class="lineno"> 1976</span> <span class="comment"> * @param training_table_meta The name of the table with the meta data. </span></div> |
| <div class="line"><a name="l01977"></a><span class="lineno"> 1977</span> <span class="comment"> * @param result_tree_table_name The name of the table where the resulting </span></div> |
| <div class="line"><a name="l01978"></a><span class="lineno"> 1978</span> <span class="comment"> * DT/RF will be stored. </span></div> |
| <div class="line"><a name="l01979"></a><span class="lineno"> 1979</span> <span class="comment"> * @param validation_table_name The validation table used for pruning tree. </span></div> |
| <div class="line"><a name="l01980"></a><span class="lineno"> 1980</span> <span class="comment"> * @param id_col_name The name of the column containing id of each point. </span></div> |
| <div class="line"><a name="l01981"></a><span class="lineno"> 1981</span> <span class="comment"> * @param class_col_name The name of the column containing correct class </span></div> |
| <div class="line"><a name="l01982"></a><span class="lineno"> 1982</span> <span class="comment"> * of each point. </span></div> |
| <div class="line"><a name="l01983"></a><span class="lineno"> 1983</span> <span class="comment"> * @param confidence_level A statistical confidence interval of the </span></div> |
| <div class="line"><a name="l01984"></a><span class="lineno"> 1984</span> <span class="comment"> * resubstitution error. </span></div> |
| <div class="line"><a name="l01985"></a><span class="lineno"> 1985</span> <span class="comment"> * @param max_tree_depth Maximum decision tree depth. </span></div> |
| <div class="line"><a name="l01986"></a><span class="lineno"> 1986</span> <span class="comment"> * @param node_prune_threshold Specifies the minimum number of samples required </span></div> |
| <div class="line"><a name="l01987"></a><span class="lineno"> 1987</span> <span class="comment"> * in a child node. </span></div> |
| <div class="line"><a name="l01988"></a><span class="lineno"> 1988</span> <span class="comment"> * @param node_split_threshold Specifies the minimum number of samples required </span></div> |
| <div class="line"><a name="l01989"></a><span class="lineno"> 1989</span> <span class="comment"> * in a node in order for a further split </span></div> |
| <div class="line"><a name="l01990"></a><span class="lineno"> 1990</span> <span class="comment"> * to be possible. </span></div> |
| <div class="line"><a name="l01991"></a><span class="lineno"> 1991</span> <span class="comment"> * @param sampling_needed Whether enabling the sampling functionality. </span></div> |
| <div class="line"><a name="l01992"></a><span class="lineno"> 1992</span> <span class="comment"> * @param h2hmv_routine_id Specifies how to handle missing values. </span></div> |
| <div class="line"><a name="l01993"></a><span class="lineno"> 1993</span> <span class="comment"> * 1 ignore, 2 explicit.</span></div> |
| <div class="line"><a name="l01994"></a><span class="lineno"> 1994</span> <span class="comment"> * @param verbosity > 0 means this function runs in verbose mode. </span></div> |
| <div class="line"><a name="l01995"></a><span class="lineno"> 1995</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l01996"></a><span class="lineno"> 1996</span> <span class="comment"> * @return The record including training related information.</span></div> |
| <div class="line"><a name="l01997"></a><span class="lineno"> 1997</span> <span class="comment"> * Details please refer to the UDT: MADLIB_SCHEMA.__train_result.</span></div> |
| <div class="line"><a name="l01998"></a><span class="lineno"> 1998</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l01999"></a><span class="lineno"> 1999</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l02000"></a><span class="lineno"> 2000</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__train_tree</div> |
| <div class="line"><a name="l02001"></a><span class="lineno"> 2001</span>  (</div> |
| <div class="line"><a name="l02002"></a><span class="lineno"> 2002</span>  split_criterion TEXT,</div> |
| <div class="line"><a name="l02003"></a><span class="lineno"> 2003</span>  num_trees INT,</div> |
| <div class="line"><a name="l02004"></a><span class="lineno"> 2004</span>  features_per_node INT,</div> |
| <div class="line"><a name="l02005"></a><span class="lineno"> 2005</span>  training_table_name TEXT, </div> |
| <div class="line"><a name="l02006"></a><span class="lineno"> 2006</span>  training_table_meta TEXT,</div> |
| <div class="line"><a name="l02007"></a><span class="lineno"> 2007</span>  result_tree_table_name TEXT,</div> |
| <div class="line"><a name="l02008"></a><span class="lineno"> 2008</span>  validation_table_name TEXT,</div> |
| <div class="line"><a name="l02009"></a><span class="lineno"> 2009</span>  id_col_name TEXT, </div> |
| <div class="line"><a name="l02010"></a><span class="lineno"> 2010</span>  class_col_name TEXT, </div> |
| <div class="line"><a name="l02011"></a><span class="lineno"> 2011</span>  confidence_level FLOAT, </div> |
| <div class="line"><a name="l02012"></a><span class="lineno"> 2012</span>  max_tree_depth INT,</div> |
| <div class="line"><a name="l02013"></a><span class="lineno"> 2013</span>  sampling_percentage FLOAT,</div> |
| <div class="line"><a name="l02014"></a><span class="lineno"> 2014</span>  node_prune_threshold FLOAT,</div> |
| <div class="line"><a name="l02015"></a><span class="lineno"> 2015</span>  node_split_threshold FLOAT,</div> |
| <div class="line"><a name="l02016"></a><span class="lineno"> 2016</span>  sampling_needed BOOLEAN,</div> |
| <div class="line"><a name="l02017"></a><span class="lineno"> 2017</span>  h2hmv_routine_id INT,</div> |
| <div class="line"><a name="l02018"></a><span class="lineno"> 2018</span>  verbosity INT</div> |
| <div class="line"><a name="l02019"></a><span class="lineno"> 2019</span>  ) </div> |
| <div class="line"><a name="l02020"></a><span class="lineno"> 2020</span> RETURNS MADLIB_SCHEMA.__train_result AS $$</div> |
| <div class="line"><a name="l02021"></a><span class="lineno"> 2021</span> DECLARE</div> |
| <div class="line"><a name="l02022"></a><span class="lineno"> 2022</span>  num_live_nodes INT;</div> |
| <div class="line"><a name="l02023"></a><span class="lineno"> 2023</span>  max_nid INT;</div> |
| <div class="line"><a name="l02024"></a><span class="lineno"> 2024</span>  location INT[];</div> |
| <div class="line"><a name="l02025"></a><span class="lineno"> 2025</span>  temp_location INT[];</div> |
| <div class="line"><a name="l02026"></a><span class="lineno"> 2026</span>  num_classes INT;</div> |
| <div class="line"><a name="l02027"></a><span class="lineno"> 2027</span>  answer record;</div> |
| <div class="line"><a name="l02028"></a><span class="lineno"> 2028</span>  location_size INT;</div> |
| <div class="line"><a name="l02029"></a><span class="lineno"> 2029</span>  begin_func_exec TIMESTAMP;</div> |
| <div class="line"><a name="l02030"></a><span class="lineno"> 2030</span>  begin_find_best TIMESTAMP;</div> |
| <div class="line"><a name="l02031"></a><span class="lineno"> 2031</span>  scv_acs_time INTERVAL;</div> |
| <div class="line"><a name="l02032"></a><span class="lineno"> 2032</span>  begin_data_transfer TIMESTAMP;</div> |
| <div class="line"><a name="l02033"></a><span class="lineno"> 2033</span>  begin_update_best TIMESTAMP;</div> |
| <div class="line"><a name="l02034"></a><span class="lineno"> 2034</span>  begin_update_child TIMESTAMP;</div> |
| <div class="line"><a name="l02035"></a><span class="lineno"> 2035</span>  begin_update_nid TIMESTAMP;</div> |
| <div class="line"><a name="l02036"></a><span class="lineno"> 2036</span>  calc_update_best INTERVAL;</div> |
| <div class="line"><a name="l02037"></a><span class="lineno"> 2037</span>  calc_update_child INTERVAL;</div> |
| <div class="line"><a name="l02038"></a><span class="lineno"> 2038</span>  calc_update_nid INTERVAL;</div> |
| <div class="line"><a name="l02039"></a><span class="lineno"> 2039</span>  ins_upd_time INTERVAL;</div> |
| <div class="line"><a name="l02040"></a><span class="lineno"> 2040</span>  begin_olap_acs TIMESTAMP;</div> |
| <div class="line"><a name="l02041"></a><span class="lineno"> 2041</span>  calc_acc_time INTERVAL;</div> |
| <div class="line"><a name="l02042"></a><span class="lineno"> 2042</span>  calc_pre_time INTERVAL;</div> |
| <div class="line"><a name="l02043"></a><span class="lineno"> 2043</span>  calc_olap_time INTERVAL;</div> |
| <div class="line"><a name="l02044"></a><span class="lineno"> 2044</span>  begin_bld_assoc TIMESTAMP;</div> |
| <div class="line"><a name="l02045"></a><span class="lineno"> 2045</span>  bld_assoc_time INTERVAL; </div> |
| <div class="line"><a name="l02046"></a><span class="lineno"> 2046</span>  begin_prune TIMESTAMP;</div> |
| <div class="line"><a name="l02047"></a><span class="lineno"> 2047</span>  prune_time INTERVAL;</div> |
| <div class="line"><a name="l02048"></a><span class="lineno"> 2048</span>  total_size FLOAT;</div> |
| <div class="line"><a name="l02049"></a><span class="lineno"> 2049</span>  sc_code INT := 1;</div> |
| <div class="line"><a name="l02050"></a><span class="lineno"> 2050</span>  curstmt TEXT := '';</div> |
| <div class="line"><a name="l02051"></a><span class="lineno"> 2051</span>  grow_tree INT := max_tree_depth;</div> |
| <div class="line"><a name="l02052"></a><span class="lineno"> 2052</span>  ret MADLIB_SCHEMA.__train_result;</div> |
| <div class="line"><a name="l02053"></a><span class="lineno"> 2053</span>  curr_level INT := 1;</div> |
| <div class="line"><a name="l02054"></a><span class="lineno"> 2054</span>  dp_ids INT[];</div> |
| <div class="line"><a name="l02055"></a><span class="lineno"> 2055</span>  dp_ids_text TEXT;</div> |
| <div class="line"><a name="l02056"></a><span class="lineno"> 2056</span>  instance_time MADLIB_SCHEMA.__gen_acc_time;</div> |
| <div class="line"><a name="l02057"></a><span class="lineno"> 2057</span>  tr_table_index INT := 1;</div> |
| <div class="line"><a name="l02058"></a><span class="lineno"> 2058</span>  tr_tables TEXT[] := '{tr_assoc_ping, tr_assoc_pong}';</div> |
| <div class="line"><a name="l02059"></a><span class="lineno"> 2059</span>  cur_tr_table TEXT := 'tr_assoc_ping';</div> |
| <div class="line"><a name="l02060"></a><span class="lineno"> 2060</span>  need_analyze BOOL := 't'::BOOL;</div> |
| <div class="line"><a name="l02061"></a><span class="lineno"> 2061</span>  attr_count INT;</div> |
| <div class="line"><a name="l02062"></a><span class="lineno"> 2062</span> BEGIN </div> |
| <div class="line"><a name="l02063"></a><span class="lineno"> 2063</span>  -- record the time costed in different steps when training</div> |
| <div class="line"><a name="l02064"></a><span class="lineno"> 2064</span>  begin_func_exec = clock_timestamp();</div> |
| <div class="line"><a name="l02065"></a><span class="lineno"> 2065</span>  scv_acs_time = begin_func_exec - begin_func_exec;</div> |
| <div class="line"><a name="l02066"></a><span class="lineno"> 2066</span>  calc_olap_time = scv_acs_time;</div> |
| <div class="line"><a name="l02067"></a><span class="lineno"> 2067</span>  calc_acc_time = scv_acs_time;</div> |
| <div class="line"><a name="l02068"></a><span class="lineno"> 2068</span>  calc_pre_time = scv_acs_time;</div> |
| <div class="line"><a name="l02069"></a><span class="lineno"> 2069</span>  ins_upd_time = scv_acs_time;</div> |
| <div class="line"><a name="l02070"></a><span class="lineno"> 2070</span>  calc_update_best = scv_acs_time;</div> |
| <div class="line"><a name="l02071"></a><span class="lineno"> 2071</span>  calc_update_child = scv_acs_time;</div> |
| <div class="line"><a name="l02072"></a><span class="lineno"> 2072</span>  calc_update_nid = scv_acs_time;</div> |
| <div class="line"><a name="l02073"></a><span class="lineno"> 2073</span>  bld_assoc_time = scv_acs_time;</div> |
| <div class="line"><a name="l02074"></a><span class="lineno"> 2074</span>  prune_time = scv_acs_time;</div> |
| <div class="line"><a name="l02075"></a><span class="lineno"> 2075</span>  </div> |
| <div class="line"><a name="l02076"></a><span class="lineno"> 2076</span>  IF(split_criterion = 'infogain') THEN</div> |
| <div class="line"><a name="l02077"></a><span class="lineno"> 2077</span>  sc_code = 1;</div> |
| <div class="line"><a name="l02078"></a><span class="lineno"> 2078</span>  ELSIF (split_criterion = 'gainratio') THEN</div> |
| <div class="line"><a name="l02079"></a><span class="lineno"> 2079</span>  sc_code = 2;</div> |
| <div class="line"><a name="l02080"></a><span class="lineno"> 2080</span>  ELSIF (split_criterion = 'gini') THEN</div> |
| <div class="line"><a name="l02081"></a><span class="lineno"> 2081</span>  sc_code = 3;</div> |
| <div class="line"><a name="l02082"></a><span class="lineno"> 2082</span>  ELSE</div> |
| <div class="line"><a name="l02083"></a><span class="lineno"> 2083</span>  RAISE EXCEPTION '%', 'Invalid split criterion!';</div> |
| <div class="line"><a name="l02084"></a><span class="lineno"> 2084</span>  END IF;</div> |
| <div class="line"><a name="l02085"></a><span class="lineno"> 2085</span> </div> |
| <div class="line"><a name="l02086"></a><span class="lineno"> 2086</span>  num_classes = MADLIB_SCHEMA.__num_of_class(training_table_meta); </div> |
| <div class="line"><a name="l02087"></a><span class="lineno"> 2087</span>  </div> |
| <div class="line"><a name="l02088"></a><span class="lineno"> 2088</span>  IF(verbosity > 0) THEN</div> |
| <div class="line"><a name="l02089"></a><span class="lineno"> 2089</span>  RAISE INFO 'NUMBER OF CLASSES IN THE TRAINING SET %', num_classes;</div> |
| <div class="line"><a name="l02090"></a><span class="lineno"> 2090</span>  END IF;</div> |
| <div class="line"><a name="l02091"></a><span class="lineno"> 2091</span>  </div> |
| <div class="line"><a name="l02092"></a><span class="lineno"> 2092</span>  IF(num_classes < 2) THEN</div> |
| <div class="line"><a name="l02093"></a><span class="lineno"> 2093</span>  RAISE EXCEPTION 'the number of classes must be greater than 2';</div> |
| <div class="line"><a name="l02094"></a><span class="lineno"> 2094</span>  END IF;</div> |
| <div class="line"><a name="l02095"></a><span class="lineno"> 2095</span> </div> |
| <div class="line"><a name="l02096"></a><span class="lineno"> 2096</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02097"></a><span class="lineno"> 2097</span>  (</div> |
| <div class="line"><a name="l02098"></a><span class="lineno"> 2098</span>  'SELECT</div> |
| <div class="line"><a name="l02099"></a><span class="lineno"> 2099</span>  count(*)</div> |
| <div class="line"><a name="l02100"></a><span class="lineno"> 2100</span>  FROM %</div> |
| <div class="line"><a name="l02101"></a><span class="lineno"> 2101</span>  WHERE column_type=''f''',</div> |
| <div class="line"><a name="l02102"></a><span class="lineno"> 2102</span>  training_table_meta</div> |
| <div class="line"><a name="l02103"></a><span class="lineno"> 2103</span>  );</div> |
| <div class="line"><a name="l02104"></a><span class="lineno"> 2104</span>  EXECUTE curstmt INTO attr_count;</div> |
| <div class="line"><a name="l02105"></a><span class="lineno"> 2105</span> </div> |
| <div class="line"><a name="l02106"></a><span class="lineno"> 2106</span>  -- generate the horizontal table for updating assinged node IDs</div> |
| <div class="line"><a name="l02107"></a><span class="lineno"> 2107</span>  PERFORM MADLIB_SCHEMA.__gen_horizontal_encoded_table</div> |
| <div class="line"><a name="l02108"></a><span class="lineno"> 2108</span>  (</div> |
| <div class="line"><a name="l02109"></a><span class="lineno"> 2109</span>  'tmp_dt_hori_table',</div> |
| <div class="line"><a name="l02110"></a><span class="lineno"> 2110</span>  training_table_name,</div> |
| <div class="line"><a name="l02111"></a><span class="lineno"> 2111</span>  attr_count,</div> |
| <div class="line"><a name="l02112"></a><span class="lineno"> 2112</span>  verbosity</div> |
| <div class="line"><a name="l02113"></a><span class="lineno"> 2113</span>  );</div> |
| <div class="line"><a name="l02114"></a><span class="lineno"> 2114</span> </div> |
| <div class="line"><a name="l02115"></a><span class="lineno"> 2115</span>  EXECUTE 'SELECT count(*) FROM tmp_dt_hori_table' INTO total_size;</div> |
| <div class="line"><a name="l02116"></a><span class="lineno"> 2116</span>  </div> |
| <div class="line"><a name="l02117"></a><span class="lineno"> 2117</span>  IF(verbosity > 0) THEN</div> |
| <div class="line"><a name="l02118"></a><span class="lineno"> 2118</span>  RAISE INFO 'INPUT TABLE SIZE: %', total_size;</div> |
| <div class="line"><a name="l02119"></a><span class="lineno"> 2119</span>  END IF;</div> |
| <div class="line"><a name="l02120"></a><span class="lineno"> 2120</span> </div> |
| <div class="line"><a name="l02121"></a><span class="lineno"> 2121</span>  begin_bld_assoc = clock_timestamp();</div> |
| <div class="line"><a name="l02122"></a><span class="lineno"> 2122</span>  cur_tr_table = tr_tables[tr_table_index];</div> |
| <div class="line"><a name="l02123"></a><span class="lineno"> 2123</span> </div> |
| <div class="line"><a name="l02124"></a><span class="lineno"> 2124</span>  -- The table of tr_assoc holds the information of which records are </div> |
| <div class="line"><a name="l02125"></a><span class="lineno"> 2125</span>  -- used during training for each tree.</div> |
| <div class="line"><a name="l02126"></a><span class="lineno"> 2126</span>  -- It has four columns.</div> |
| <div class="line"><a name="l02127"></a><span class="lineno"> 2127</span>  -- id -- The id of one record.</div> |
| <div class="line"><a name="l02128"></a><span class="lineno"> 2128</span>  -- tid -- The id of a tree.</div> |
| <div class="line"><a name="l02129"></a><span class="lineno"> 2129</span>  -- nid -- The id of a node in a tree.</div> |
| <div class="line"><a name="l02130"></a><span class="lineno"> 2130</span>  -- weight -- The times a record is assigned to a node.</div> |
| <div class="line"><a name="l02131"></a><span class="lineno"> 2131</span>  IF (sampling_needed) THEN</div> |
| <div class="line"><a name="l02132"></a><span class="lineno"> 2132</span>  PERFORM MADLIB_SCHEMA.__sample_with_replacement</div> |
| <div class="line"><a name="l02133"></a><span class="lineno"> 2133</span>  (</div> |
| <div class="line"><a name="l02134"></a><span class="lineno"> 2134</span>  num_trees,</div> |
| <div class="line"><a name="l02135"></a><span class="lineno"> 2135</span>  round(sampling_percentage * total_size)::INT,</div> |
| <div class="line"><a name="l02136"></a><span class="lineno"> 2136</span>  'tmp_dt_hori_table',</div> |
| <div class="line"><a name="l02137"></a><span class="lineno"> 2137</span>  cur_tr_table</div> |
| <div class="line"><a name="l02138"></a><span class="lineno"> 2138</span>  );</div> |
| <div class="line"><a name="l02139"></a><span class="lineno"> 2139</span>  ELSE</div> |
| <div class="line"><a name="l02140"></a><span class="lineno"> 2140</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02141"></a><span class="lineno"> 2141</span>  (</div> |
| <div class="line"><a name="l02142"></a><span class="lineno"> 2142</span>  'INSERT INTO % </div> |
| <div class="line"><a name="l02143"></a><span class="lineno"> 2143</span>  SELECT id, 1 as tid, 1 as nid, 1 as weight </div> |
| <div class="line"><a name="l02144"></a><span class="lineno"> 2144</span>  FROM %',</div> |
| <div class="line"><a name="l02145"></a><span class="lineno"> 2145</span>  ARRAY[</div> |
| <div class="line"><a name="l02146"></a><span class="lineno"> 2146</span>  cur_tr_table,</div> |
| <div class="line"><a name="l02147"></a><span class="lineno"> 2147</span>  'tmp_dt_hori_table'</div> |
| <div class="line"><a name="l02148"></a><span class="lineno"> 2148</span>  ]</div> |
| <div class="line"><a name="l02149"></a><span class="lineno"> 2149</span>  );</div> |
| <div class="line"><a name="l02150"></a><span class="lineno"> 2150</span>  EXECUTE curstmt; </div> |
| <div class="line"><a name="l02151"></a><span class="lineno"> 2151</span>  END IF;</div> |
| <div class="line"><a name="l02152"></a><span class="lineno"> 2152</span>  </div> |
| <div class="line"><a name="l02153"></a><span class="lineno"> 2153</span>  -- analyze ping</div> |
| <div class="line"><a name="l02154"></a><span class="lineno"> 2154</span>  EXECUTE 'ANALYZE ' || cur_tr_table;</div> |
| <div class="line"><a name="l02155"></a><span class="lineno"> 2155</span>  bld_assoc_time = clock_timestamp() - begin_bld_assoc;</div> |
| <div class="line"><a name="l02156"></a><span class="lineno"> 2156</span> </div> |
| <div class="line"><a name="l02157"></a><span class="lineno"> 2157</span>  -- generate the root node for all trees.</div> |
| <div class="line"><a name="l02158"></a><span class="lineno"> 2158</span>  -- the generated numbers are the same for the two generate_series</div> |
| <div class="line"><a name="l02159"></a><span class="lineno"> 2159</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02160"></a><span class="lineno"> 2160</span>  (</div> |
| <div class="line"><a name="l02161"></a><span class="lineno"> 2161</span>  'INSERT INTO % </div> |
| <div class="line"><a name="l02162"></a><span class="lineno"> 2162</span>  (id, tree_location, feature, probability, max_class,scv, </div> |
| <div class="line"><a name="l02163"></a><span class="lineno"> 2163</span>  live, num_of_samples, parent_id, tid) </div> |
| <div class="line"><a name="l02164"></a><span class="lineno"> 2164</span>  SELECT generate_series(1, %), ARRAY[0], 0, 1, 1, 1, 1, 0, 0, </div> |
| <div class="line"><a name="l02165"></a><span class="lineno"> 2165</span>  generate_series(1, %)',</div> |
| <div class="line"><a name="l02166"></a><span class="lineno"> 2166</span>  ARRAY[</div> |
| <div class="line"><a name="l02167"></a><span class="lineno"> 2167</span>  result_tree_table_name,</div> |
| <div class="line"><a name="l02168"></a><span class="lineno"> 2168</span>  num_trees::TEXT,</div> |
| <div class="line"><a name="l02169"></a><span class="lineno"> 2169</span>  num_trees::TEXT</div> |
| <div class="line"><a name="l02170"></a><span class="lineno"> 2170</span>  ]</div> |
| <div class="line"><a name="l02171"></a><span class="lineno"> 2171</span>  ) INTO curstmt;</div> |
| <div class="line"><a name="l02172"></a><span class="lineno"> 2172</span>  </div> |
| <div class="line"><a name="l02173"></a><span class="lineno"> 2173</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l02174"></a><span class="lineno"> 2174</span>  </div> |
| <div class="line"><a name="l02175"></a><span class="lineno"> 2175</span>  max_nid = num_trees;</div> |
| <div class="line"><a name="l02176"></a><span class="lineno"> 2176</span>  location_size = 0;</div> |
| <div class="line"><a name="l02177"></a><span class="lineno"> 2177</span> </div> |
| <div class="line"><a name="l02178"></a><span class="lineno"> 2178</span> </div> |
| <div class="line"><a name="l02179"></a><span class="lineno"> 2179</span>  LOOP</div> |
| <div class="line"><a name="l02180"></a><span class="lineno"> 2180</span>  EXECUTE 'SELECT COUNT(id) FROM ' || result_tree_table_name || </div> |
| <div class="line"><a name="l02181"></a><span class="lineno"> 2181</span>  ' WHERE live > 0 AND array_upper(tree_location,1)='||</div> |
| <div class="line"><a name="l02182"></a><span class="lineno"> 2182</span>  curr_level||';' INTO num_live_nodes;</div> |
| <div class="line"><a name="l02183"></a><span class="lineno"> 2183</span>  </div> |
| <div class="line"><a name="l02184"></a><span class="lineno"> 2184</span>  IF (num_live_nodes < 1) THEN</div> |
| <div class="line"><a name="l02185"></a><span class="lineno"> 2185</span>  IF(verbosity > 0) THEN</div> |
| <div class="line"><a name="l02186"></a><span class="lineno"> 2186</span>  RAISE INFO 'EXIT: %', 'no live nodes to split';</div> |
| <div class="line"><a name="l02187"></a><span class="lineno"> 2187</span>  END IF;</div> |
| <div class="line"><a name="l02188"></a><span class="lineno"> 2188</span>  </div> |
| <div class="line"><a name="l02189"></a><span class="lineno"> 2189</span>  EXIT;</div> |
| <div class="line"><a name="l02190"></a><span class="lineno"> 2190</span>  END IF;</div> |
| <div class="line"><a name="l02191"></a><span class="lineno"> 2191</span> </div> |
| <div class="line"><a name="l02192"></a><span class="lineno"> 2192</span>  IF (verbosity > 0) THEN</div> |
| <div class="line"><a name="l02193"></a><span class="lineno"> 2193</span>  RAISE INFO 'Running on level:%', curr_level;</div> |
| <div class="line"><a name="l02194"></a><span class="lineno"> 2194</span>  END IF;</div> |
| <div class="line"><a name="l02195"></a><span class="lineno"> 2195</span>  </div> |
| <div class="line"><a name="l02196"></a><span class="lineno"> 2196</span>  begin_olap_acs = clock_timestamp();</div> |
| <div class="line"><a name="l02197"></a><span class="lineno"> 2197</span>  </div> |
| <div class="line"><a name="l02198"></a><span class="lineno"> 2198</span>  instance_time = MADLIB_SCHEMA.__gen_acc</div> |
| <div class="line"><a name="l02199"></a><span class="lineno"> 2199</span>  (</div> |
| <div class="line"><a name="l02200"></a><span class="lineno"> 2200</span>  training_table_name,</div> |
| <div class="line"><a name="l02201"></a><span class="lineno"> 2201</span>  training_table_meta,</div> |
| <div class="line"><a name="l02202"></a><span class="lineno"> 2202</span>  result_tree_table_name,</div> |
| <div class="line"><a name="l02203"></a><span class="lineno"> 2203</span>  cur_tr_table,</div> |
| <div class="line"><a name="l02204"></a><span class="lineno"> 2204</span>  'sf_assoc',</div> |
| <div class="line"><a name="l02205"></a><span class="lineno"> 2205</span>  features_per_node,</div> |
| <div class="line"><a name="l02206"></a><span class="lineno"> 2206</span>  num_classes,</div> |
| <div class="line"><a name="l02207"></a><span class="lineno"> 2207</span>  sampling_needed,</div> |
| <div class="line"><a name="l02208"></a><span class="lineno"> 2208</span>  verbosity</div> |
| <div class="line"><a name="l02209"></a><span class="lineno"> 2209</span>  );</div> |
| <div class="line"><a name="l02210"></a><span class="lineno"> 2210</span> </div> |
| <div class="line"><a name="l02211"></a><span class="lineno"> 2211</span>  IF (h2hmv_routine_id=1) THEN</div> |
| <div class="line"><a name="l02212"></a><span class="lineno"> 2212</span>  -- For ignore, we need the true size of nodes to handle the</div> |
| <div class="line"><a name="l02213"></a><span class="lineno"> 2213</span>  -- missing values.</div> |
| <div class="line"><a name="l02214"></a><span class="lineno"> 2214</span>  TRUNCATE node_size_aux;</div> |
| <div class="line"><a name="l02215"></a><span class="lineno"> 2215</span>  </div> |
| <div class="line"><a name="l02216"></a><span class="lineno"> 2216</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02217"></a><span class="lineno"> 2217</span>  (</div> |
| <div class="line"><a name="l02218"></a><span class="lineno"> 2218</span>  'INSERT INTO node_size_aux </div> |
| <div class="line"><a name="l02219"></a><span class="lineno"> 2219</span>  SELECT tr.tid, tr.nid, sum(weight) as count</div> |
| <div class="line"><a name="l02220"></a><span class="lineno"> 2220</span>  FROM % tr</div> |
| <div class="line"><a name="l02221"></a><span class="lineno"> 2221</span>  GROUP BY tr.tid, tr.nid',</div> |
| <div class="line"><a name="l02222"></a><span class="lineno"> 2222</span>  cur_tr_table</div> |
| <div class="line"><a name="l02223"></a><span class="lineno"> 2223</span>  );</div> |
| <div class="line"><a name="l02224"></a><span class="lineno"> 2224</span> </div> |
| <div class="line"><a name="l02225"></a><span class="lineno"> 2225</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l02226"></a><span class="lineno"> 2226</span>  END IF;</div> |
| <div class="line"><a name="l02227"></a><span class="lineno"> 2227</span> </div> |
| <div class="line"><a name="l02228"></a><span class="lineno"> 2228</span>  calc_pre_time = calc_pre_time + instance_time.calc_pre_time;</div> |
| <div class="line"><a name="l02229"></a><span class="lineno"> 2229</span>  calc_acc_time = calc_acc_time + instance_time.calc_acc_time;</div> |
| <div class="line"><a name="l02230"></a><span class="lineno"> 2230</span>  calc_olap_time = calc_olap_time + (clock_timestamp() - begin_olap_acs);</div> |
| <div class="line"><a name="l02231"></a><span class="lineno"> 2231</span> </div> |
| <div class="line"><a name="l02232"></a><span class="lineno"> 2232</span>  curr_level = curr_level + 1;</div> |
| <div class="line"><a name="l02233"></a><span class="lineno"> 2233</span>  </div> |
| <div class="line"><a name="l02234"></a><span class="lineno"> 2234</span>  begin_find_best = clock_timestamp();</div> |
| <div class="line"><a name="l02235"></a><span class="lineno"> 2235</span> </div> |
| <div class="line"><a name="l02236"></a><span class="lineno"> 2236</span>  PERFORM MADLIB_SCHEMA.__find_best_split</div> |
| <div class="line"><a name="l02237"></a><span class="lineno"> 2237</span>  (</div> |
| <div class="line"><a name="l02238"></a><span class="lineno"> 2238</span>  'training_instance',</div> |
| <div class="line"><a name="l02239"></a><span class="lineno"> 2239</span>  confidence_level,</div> |
| <div class="line"><a name="l02240"></a><span class="lineno"> 2240</span>  training_table_meta,</div> |
| <div class="line"><a name="l02241"></a><span class="lineno"> 2241</span>  sc_code,</div> |
| <div class="line"><a name="l02242"></a><span class="lineno"> 2242</span>  grow_tree,</div> |
| <div class="line"><a name="l02243"></a><span class="lineno"> 2243</span>  'find_best_answer_table',</div> |
| <div class="line"><a name="l02244"></a><span class="lineno"> 2244</span>  h2hmv_routine_id,</div> |
| <div class="line"><a name="l02245"></a><span class="lineno"> 2245</span>  num_classes</div> |
| <div class="line"><a name="l02246"></a><span class="lineno"> 2246</span>  );</div> |
| <div class="line"><a name="l02247"></a><span class="lineno"> 2247</span>  IF (verbosity > 0) THEN</div> |
| <div class="line"><a name="l02248"></a><span class="lineno"> 2248</span>  RAISE INFO 'find best time at this level:%', </div> |
| <div class="line"><a name="l02249"></a><span class="lineno"> 2249</span>  clock_timestamp() - begin_find_best;</div> |
| <div class="line"><a name="l02250"></a><span class="lineno"> 2250</span>  END IF;</div> |
| <div class="line"><a name="l02251"></a><span class="lineno"> 2251</span>  grow_tree = grow_tree - 1;</div> |
| <div class="line"><a name="l02252"></a><span class="lineno"> 2252</span>  </div> |
| <div class="line"><a name="l02253"></a><span class="lineno"> 2253</span>  scv_acs_time = scv_acs_time + </div> |
| <div class="line"><a name="l02254"></a><span class="lineno"> 2254</span>  (clock_timestamp() - begin_find_best);</div> |
| <div class="line"><a name="l02255"></a><span class="lineno"> 2255</span>  begin_data_transfer = clock_timestamp();</div> |
| <div class="line"><a name="l02256"></a><span class="lineno"> 2256</span>  begin_update_best = clock_timestamp();</div> |
| <div class="line"><a name="l02257"></a><span class="lineno"> 2257</span>  </div> |
| <div class="line"><a name="l02258"></a><span class="lineno"> 2258</span>  -- We get the calculation result for current level. </div> |
| <div class="line"><a name="l02259"></a><span class="lineno"> 2259</span>  -- Update the nodes of previous level firstly.</div> |
| <div class="line"><a name="l02260"></a><span class="lineno"> 2260</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02261"></a><span class="lineno"> 2261</span>  (</div> |
| <div class="line"><a name="l02262"></a><span class="lineno"> 2262</span>  'UPDATE % t </div> |
| <div class="line"><a name="l02263"></a><span class="lineno"> 2263</span>  SET feature = c.feature,</div> |
| <div class="line"><a name="l02264"></a><span class="lineno"> 2264</span>  probability = c.probability,</div> |
| <div class="line"><a name="l02265"></a><span class="lineno"> 2265</span>  max_class = c.max_class,</div> |
| <div class="line"><a name="l02266"></a><span class="lineno"> 2266</span>  scv = c.max_scv,</div> |
| <div class="line"><a name="l02267"></a><span class="lineno"> 2267</span>  ebp_coeff = c.ebp_coeff,</div> |
| <div class="line"><a name="l02268"></a><span class="lineno"> 2268</span>  num_of_samples = c.node_size,</div> |
| <div class="line"><a name="l02269"></a><span class="lineno"> 2269</span>  live = 0,</div> |
| <div class="line"><a name="l02270"></a><span class="lineno"> 2270</span>  is_cont = c.is_cont,</div> |
| <div class="line"><a name="l02271"></a><span class="lineno"> 2271</span>  split_value = c.split_value </div> |
| <div class="line"><a name="l02272"></a><span class="lineno"> 2272</span>  FROM find_best_answer_table c</div> |
| <div class="line"><a name="l02273"></a><span class="lineno"> 2273</span>  WHERE t.id=c.node_id AND t.tid=c.tid',</div> |
| <div class="line"><a name="l02274"></a><span class="lineno"> 2274</span>  ARRAY[</div> |
| <div class="line"><a name="l02275"></a><span class="lineno"> 2275</span>  result_tree_table_name::TEXT</div> |
| <div class="line"><a name="l02276"></a><span class="lineno"> 2276</span>  ]</div> |
| <div class="line"><a name="l02277"></a><span class="lineno"> 2277</span>  ) INTO curstmt;</div> |
| <div class="line"><a name="l02278"></a><span class="lineno"> 2278</span>  </div> |
| <div class="line"><a name="l02279"></a><span class="lineno"> 2279</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l02280"></a><span class="lineno"> 2280</span>  </div> |
| <div class="line"><a name="l02281"></a><span class="lineno"> 2281</span>  calc_update_best = calc_update_best + </div> |
| <div class="line"><a name="l02282"></a><span class="lineno"> 2282</span>  (clock_timestamp() - begin_update_best);</div> |
| <div class="line"><a name="l02283"></a><span class="lineno"> 2283</span>  begin_update_child = clock_timestamp();</div> |
| <div class="line"><a name="l02284"></a><span class="lineno"> 2284</span> </div> |
| <div class="line"><a name="l02285"></a><span class="lineno"> 2285</span>  curstmt=</div> |
| <div class="line"><a name="l02286"></a><span class="lineno"> 2286</span>  MADLIB_SCHEMA.__format(</div> |
| <div class="line"><a name="l02287"></a><span class="lineno"> 2287</span>  'INSERT INTO %(id, tree_location, feature, probability, </div> |
| <div class="line"><a name="l02288"></a><span class="lineno"> 2288</span>  max_class, scv, live, parent_id, tid, dp_ids)</div> |
| <div class="line"><a name="l02289"></a><span class="lineno"> 2289</span>  SELECT %+row, array_append(tree_location, fval),</div> |
| <div class="line"><a name="l02290"></a><span class="lineno"> 2290</span>  0, 1, 1, 1, %, ans.node_id, ans.tid, </div> |
| <div class="line"><a name="l02291"></a><span class="lineno"> 2291</span>  CASE when(NOT ans.is_cont) then</div> |
| <div class="line"><a name="l02292"></a><span class="lineno"> 2292</span>  array_append( dp_ids, ans.feature)</div> |
| <div class="line"><a name="l02293"></a><span class="lineno"> 2293</span>  ELSE</div> |
| <div class="line"><a name="l02294"></a><span class="lineno"> 2294</span>  dp_ids</div> |
| <div class="line"><a name="l02295"></a><span class="lineno"> 2295</span>  END</div> |
| <div class="line"><a name="l02296"></a><span class="lineno"> 2296</span>  FROM % tree,</div> |
| <div class="line"><a name="l02297"></a><span class="lineno"> 2297</span>  (</div> |
| <div class="line"><a name="l02298"></a><span class="lineno"> 2298</span>  SELECT *, </div> |
| <div class="line"><a name="l02299"></a><span class="lineno"> 2299</span>  row_number() </div> |
| <div class="line"><a name="l02300"></a><span class="lineno"> 2300</span>  OVER (ORDER BY l.tid, l.node_id, l.fval) AS row</div> |
| <div class="line"><a name="l02301"></a><span class="lineno"> 2301</span>  FROM</div> |
| <div class="line"><a name="l02302"></a><span class="lineno"> 2302</span>  (</div> |
| <div class="line"><a name="l02303"></a><span class="lineno"> 2303</span>  SELECT *, </div> |
| <div class="line"><a name="l02304"></a><span class="lineno"> 2304</span>  CASE WHEN (is_cont) THEN</div> |
| <div class="line"><a name="l02305"></a><span class="lineno"> 2305</span>  generate_series(1,2)</div> |
| <div class="line"><a name="l02306"></a><span class="lineno"> 2306</span>  ELSE</div> |
| <div class="line"><a name="l02307"></a><span class="lineno"> 2307</span>  generate_series(1, distinct_features)</div> |
| <div class="line"><a name="l02308"></a><span class="lineno"> 2308</span>  END AS fval </div> |
| <div class="line"><a name="l02309"></a><span class="lineno"> 2309</span>  FROM</div> |
| <div class="line"><a name="l02310"></a><span class="lineno"> 2310</span>  find_best_answer_table</div> |
| <div class="line"><a name="l02311"></a><span class="lineno"> 2311</span>  WHERE live>0 AND coalesce(feature, 0) <> 0 </div> |
| <div class="line"><a name="l02312"></a><span class="lineno"> 2312</span>  AND node_size >= % AND node_size >= %</div> |
| <div class="line"><a name="l02313"></a><span class="lineno"> 2313</span>  ) l</div> |
| <div class="line"><a name="l02314"></a><span class="lineno"> 2314</span>  ) ans</div> |
| <div class="line"><a name="l02315"></a><span class="lineno"> 2315</span>  WHERE tree.id=ans.node_id and tree.tid=ans.tid;',</div> |
| <div class="line"><a name="l02316"></a><span class="lineno"> 2316</span>  ARRAY[</div> |
| <div class="line"><a name="l02317"></a><span class="lineno"> 2317</span>  result_tree_table_name,</div> |
| <div class="line"><a name="l02318"></a><span class="lineno"> 2318</span>  (max_nid)::TEXT,</div> |
| <div class="line"><a name="l02319"></a><span class="lineno"> 2319</span>  curr_level::TEXT,</div> |
| <div class="line"><a name="l02320"></a><span class="lineno"> 2320</span>  result_tree_table_name,</div> |
| <div class="line"><a name="l02321"></a><span class="lineno"> 2321</span>  (total_size * node_prune_threshold)::TEXT,</div> |
| <div class="line"><a name="l02322"></a><span class="lineno"> 2322</span>  (total_size * node_split_threshold)::TEXT </div> |
| <div class="line"><a name="l02323"></a><span class="lineno"> 2323</span>  ]</div> |
| <div class="line"><a name="l02324"></a><span class="lineno"> 2324</span>  );</div> |
| <div class="line"><a name="l02325"></a><span class="lineno"> 2325</span>  IF(verbosity > 0) THEN</div> |
| <div class="line"><a name="l02326"></a><span class="lineno"> 2326</span>  RAISE INFO 'Generate Child Nodes:%', curstmt;</div> |
| <div class="line"><a name="l02327"></a><span class="lineno"> 2327</span>  END IF;</div> |
| <div class="line"><a name="l02328"></a><span class="lineno"> 2328</span> </div> |
| <div class="line"><a name="l02329"></a><span class="lineno"> 2329</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l02330"></a><span class="lineno"> 2330</span>  </div> |
| <div class="line"><a name="l02331"></a><span class="lineno"> 2331</span>  EXECUTE 'SELECT max(id) FROM '||result_tree_table_name INTO max_nid;</div> |
| <div class="line"><a name="l02332"></a><span class="lineno"> 2332</span> </div> |
| <div class="line"><a name="l02333"></a><span class="lineno"> 2333</span>  IF(verbosity > 0) THEN</div> |
| <div class="line"><a name="l02334"></a><span class="lineno"> 2334</span>  RAISE INFO 'Max nid:%, level:%', max_nid, curr_level;</div> |
| <div class="line"><a name="l02335"></a><span class="lineno"> 2335</span>  END IF;</div> |
| <div class="line"><a name="l02336"></a><span class="lineno"> 2336</span> </div> |
| <div class="line"><a name="l02337"></a><span class="lineno"> 2337</span>  -- insert the leftmost child node id and relevant info</div> |
| <div class="line"><a name="l02338"></a><span class="lineno"> 2338</span>  -- to the assoc_aux table, so that we will make use of this </div> |
| <div class="line"><a name="l02339"></a><span class="lineno"> 2339</span>  -- info to update the assigned nid the samples belong to</div> |
| <div class="line"><a name="l02340"></a><span class="lineno"> 2340</span>  -- the current node whose id is answer.node_id.</div> |
| <div class="line"><a name="l02341"></a><span class="lineno"> 2341</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02342"></a><span class="lineno"> 2342</span>  (</div> |
| <div class="line"><a name="l02343"></a><span class="lineno"> 2343</span>  'INSERT INTO assoc_aux</div> |
| <div class="line"><a name="l02344"></a><span class="lineno"> 2344</span>  (nid, fid, lmc_id, svalue, is_cont)</div> |
| <div class="line"><a name="l02345"></a><span class="lineno"> 2345</span>  SELECT t.id, t.feature, min(l.id), </div> |
| <div class="line"><a name="l02346"></a><span class="lineno"> 2346</span>  t.split_value, t.is_cont</div> |
| <div class="line"><a name="l02347"></a><span class="lineno"> 2347</span>  FROM</div> |
| <div class="line"><a name="l02348"></a><span class="lineno"> 2348</span>  (SELECT id, parent_id </div> |
| <div class="line"><a name="l02349"></a><span class="lineno"> 2349</span>  FROM % </div> |
| <div class="line"><a name="l02350"></a><span class="lineno"> 2350</span>  WHERE array_upper(tree_location,1)=%) l,</div> |
| <div class="line"><a name="l02351"></a><span class="lineno"> 2351</span>  % t</div> |
| <div class="line"><a name="l02352"></a><span class="lineno"> 2352</span>  WHERE l.parent_id=t.id</div> |
| <div class="line"><a name="l02353"></a><span class="lineno"> 2353</span>  GROUP BY t.id, t.feature, t.split_value, t.is_cont;',</div> |
| <div class="line"><a name="l02354"></a><span class="lineno"> 2354</span>  ARRAY[</div> |
| <div class="line"><a name="l02355"></a><span class="lineno"> 2355</span>  result_tree_table_name,</div> |
| <div class="line"><a name="l02356"></a><span class="lineno"> 2356</span>  curr_level::TEXT,</div> |
| <div class="line"><a name="l02357"></a><span class="lineno"> 2357</span>  result_tree_table_name</div> |
| <div class="line"><a name="l02358"></a><span class="lineno"> 2358</span>  ]</div> |
| <div class="line"><a name="l02359"></a><span class="lineno"> 2359</span>  ) INTO curstmt;</div> |
| <div class="line"><a name="l02360"></a><span class="lineno"> 2360</span> </div> |
| <div class="line"><a name="l02361"></a><span class="lineno"> 2361</span>  IF(verbosity > 0) THEN</div> |
| <div class="line"><a name="l02362"></a><span class="lineno"> 2362</span>  RAISE INFO 'Update lmc_child Info:%', curstmt;</div> |
| <div class="line"><a name="l02363"></a><span class="lineno"> 2363</span>  END IF;</div> |
| <div class="line"><a name="l02364"></a><span class="lineno"> 2364</span> </div> |
| <div class="line"><a name="l02365"></a><span class="lineno"> 2365</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l02366"></a><span class="lineno"> 2366</span> </div> |
| <div class="line"><a name="l02367"></a><span class="lineno"> 2367</span>  -- delete the unused nodes on the previous level</div> |
| <div class="line"><a name="l02368"></a><span class="lineno"> 2368</span>  -- delete those nodes with a size less than node_prune_threshold</div> |
| <div class="line"><a name="l02369"></a><span class="lineno"> 2369</span>  -- node_prune_threshold will not apply to root node, </div> |
| <div class="line"><a name="l02370"></a><span class="lineno"> 2370</span>  -- the level is 1 (curr_level - 1 = 1);</div> |
| <div class="line"><a name="l02371"></a><span class="lineno"> 2371</span>  IF (curr_level > 2) THEN</div> |
| <div class="line"><a name="l02372"></a><span class="lineno"> 2372</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02373"></a><span class="lineno"> 2373</span>  (</div> |
| <div class="line"><a name="l02374"></a><span class="lineno"> 2374</span>  'DELETE FROM % t </div> |
| <div class="line"><a name="l02375"></a><span class="lineno"> 2375</span>  WHERE t.num_of_samples < % OR live = %;',</div> |
| <div class="line"><a name="l02376"></a><span class="lineno"> 2376</span>  ARRAY[</div> |
| <div class="line"><a name="l02377"></a><span class="lineno"> 2377</span>  result_tree_table_name::TEXT,</div> |
| <div class="line"><a name="l02378"></a><span class="lineno"> 2378</span>  (total_size * node_prune_threshold)::TEXT,</div> |
| <div class="line"><a name="l02379"></a><span class="lineno"> 2379</span>  (curr_level - 1)::TEXT</div> |
| <div class="line"><a name="l02380"></a><span class="lineno"> 2380</span>  ]</div> |
| <div class="line"><a name="l02381"></a><span class="lineno"> 2381</span>  );</div> |
| <div class="line"><a name="l02382"></a><span class="lineno"> 2382</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l02383"></a><span class="lineno"> 2383</span>  END IF;</div> |
| <div class="line"><a name="l02384"></a><span class="lineno"> 2384</span>  </div> |
| <div class="line"><a name="l02385"></a><span class="lineno"> 2385</span>  calc_update_child = calc_update_child + (clock_timestamp() - begin_update_child);</div> |
| <div class="line"><a name="l02386"></a><span class="lineno"> 2386</span>  begin_update_nid = clock_timestamp();</div> |
| <div class="line"><a name="l02387"></a><span class="lineno"> 2387</span>  </div> |
| <div class="line"><a name="l02388"></a><span class="lineno"> 2388</span>  -- update the assigned node id for each sample on the current level</div> |
| <div class="line"><a name="l02389"></a><span class="lineno"> 2389</span>  tr_table_index = (tr_table_index % 2) + 1; </div> |
| <div class="line"><a name="l02390"></a><span class="lineno"> 2390</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02391"></a><span class="lineno"> 2391</span>  (</div> |
| <div class="line"><a name="l02392"></a><span class="lineno"> 2392</span>  'INSERT INTO % (id, nid, tid, weight)</div> |
| <div class="line"><a name="l02393"></a><span class="lineno"> 2393</span>  SELECT </div> |
| <div class="line"><a name="l02394"></a><span class="lineno"> 2394</span>  tr.id,</div> |
| <div class="line"><a name="l02395"></a><span class="lineno"> 2395</span>  au.lmc_id - 1 +</div> |
| <div class="line"><a name="l02396"></a><span class="lineno"> 2396</span>  CASE WHEN (au.is_cont) THEN</div> |
| <div class="line"><a name="l02397"></a><span class="lineno"> 2397</span>  CASE WHEN (svalue < vt.fvals[au.fid]) THEN</div> |
| <div class="line"><a name="l02398"></a><span class="lineno"> 2398</span>  2</div> |
| <div class="line"><a name="l02399"></a><span class="lineno"> 2399</span>  ELSE</div> |
| <div class="line"><a name="l02400"></a><span class="lineno"> 2400</span>  1</div> |
| <div class="line"><a name="l02401"></a><span class="lineno"> 2401</span>  END</div> |
| <div class="line"><a name="l02402"></a><span class="lineno"> 2402</span>  ELSE</div> |
| <div class="line"><a name="l02403"></a><span class="lineno"> 2403</span>  vt.fvals[au.fid]::INT</div> |
| <div class="line"><a name="l02404"></a><span class="lineno"> 2404</span>  END AS nid,</div> |
| <div class="line"><a name="l02405"></a><span class="lineno"> 2405</span>  tid, weight</div> |
| <div class="line"><a name="l02406"></a><span class="lineno"> 2406</span>  FROM % tr, % vt, assoc_aux au</div> |
| <div class="line"><a name="l02407"></a><span class="lineno"> 2407</span>  WHERE tr.nid = au.nid AND vt.id = tr.id AND vt.fvals[au.fid] IS NOT NULL',</div> |
| <div class="line"><a name="l02408"></a><span class="lineno"> 2408</span>  ARRAY[</div> |
| <div class="line"><a name="l02409"></a><span class="lineno"> 2409</span>  tr_tables[tr_table_index],</div> |
| <div class="line"><a name="l02410"></a><span class="lineno"> 2410</span>  cur_tr_table,</div> |
| <div class="line"><a name="l02411"></a><span class="lineno"> 2411</span>  'tmp_dt_hori_table'</div> |
| <div class="line"><a name="l02412"></a><span class="lineno"> 2412</span>  ]</div> |
| <div class="line"><a name="l02413"></a><span class="lineno"> 2413</span>  ); </div> |
| <div class="line"><a name="l02414"></a><span class="lineno"> 2414</span>  IF (verbosity > 0) THEN</div> |
| <div class="line"><a name="l02415"></a><span class="lineno"> 2415</span>  RAISE INFO '%', curstmt;</div> |
| <div class="line"><a name="l02416"></a><span class="lineno"> 2416</span>  END IF;</div> |
| <div class="line"><a name="l02417"></a><span class="lineno"> 2417</span> </div> |
| <div class="line"><a name="l02418"></a><span class="lineno"> 2418</span>  EXECUTE curstmt; </div> |
| <div class="line"><a name="l02419"></a><span class="lineno"> 2419</span>  EXECUTE 'TRUNCATE ' || cur_tr_table; </div> |
| <div class="line"><a name="l02420"></a><span class="lineno"> 2420</span>  cur_tr_table = tr_tables[tr_table_index];</div> |
| <div class="line"><a name="l02421"></a><span class="lineno"> 2421</span> </div> |
| <div class="line"><a name="l02422"></a><span class="lineno"> 2422</span>  IF (need_analyze) THEN</div> |
| <div class="line"><a name="l02423"></a><span class="lineno"> 2423</span>  -- analyze pong table</div> |
| <div class="line"><a name="l02424"></a><span class="lineno"> 2424</span>  EXECUTE 'ANALYZE ' || cur_tr_table;</div> |
| <div class="line"><a name="l02425"></a><span class="lineno"> 2425</span>  need_analyze = 'f'::BOOL;</div> |
| <div class="line"><a name="l02426"></a><span class="lineno"> 2426</span>  END IF;</div> |
| <div class="line"><a name="l02427"></a><span class="lineno"> 2427</span> </div> |
| <div class="line"><a name="l02428"></a><span class="lineno"> 2428</span>  EXECUTE 'TRUNCATE assoc_aux';</div> |
| <div class="line"><a name="l02429"></a><span class="lineno"> 2429</span>  </div> |
| <div class="line"><a name="l02430"></a><span class="lineno"> 2430</span>  calc_update_nid = calc_update_nid + (clock_timestamp() - begin_update_nid);</div> |
| <div class="line"><a name="l02431"></a><span class="lineno"> 2431</span>  </div> |
| <div class="line"><a name="l02432"></a><span class="lineno"> 2432</span>  ins_upd_time = ins_upd_time + </div> |
| <div class="line"><a name="l02433"></a><span class="lineno"> 2433</span>  (clock_timestamp() - begin_data_transfer); </div> |
| <div class="line"><a name="l02434"></a><span class="lineno"> 2434</span>  IF(verbosity > 0) THEN</div> |
| <div class="line"><a name="l02435"></a><span class="lineno"> 2435</span>  RAISE INFO 'computation time in this level:%',</div> |
| <div class="line"><a name="l02436"></a><span class="lineno"> 2436</span>  (clock_timestamp() - begin_find_best);</div> |
| <div class="line"><a name="l02437"></a><span class="lineno"> 2437</span>  END IF;</div> |
| <div class="line"><a name="l02438"></a><span class="lineno"> 2438</span>  </div> |
| <div class="line"><a name="l02439"></a><span class="lineno"> 2439</span>  END LOOP;</div> |
| <div class="line"><a name="l02440"></a><span class="lineno"> 2440</span>  </div> |
| <div class="line"><a name="l02441"></a><span class="lineno"> 2441</span>  PERFORM MADLIB_SCHEMA.__generate_final_tree(result_tree_table_name);</div> |
| <div class="line"><a name="l02442"></a><span class="lineno"> 2442</span>  </div> |
| <div class="line"><a name="l02443"></a><span class="lineno"> 2443</span>  begin_prune = clock_timestamp();</div> |
| <div class="line"><a name="l02444"></a><span class="lineno"> 2444</span>  IF (confidence_level < 100.0) THEN</div> |
| <div class="line"><a name="l02445"></a><span class="lineno"> 2445</span>  PERFORM MADLIB_SCHEMA.__ebp_prune_tree(result_tree_table_name);</div> |
| <div class="line"><a name="l02446"></a><span class="lineno"> 2446</span>  END IF;</div> |
| <div class="line"><a name="l02447"></a><span class="lineno"> 2447</span>  </div> |
| <div class="line"><a name="l02448"></a><span class="lineno"> 2448</span>  IF (validation_table_name IS NOT NULL) THEN</div> |
| <div class="line"><a name="l02449"></a><span class="lineno"> 2449</span>  PERFORM MADLIB_SCHEMA.__rep_prune_tree</div> |
| <div class="line"><a name="l02450"></a><span class="lineno"> 2450</span>  (</div> |
| <div class="line"><a name="l02451"></a><span class="lineno"> 2451</span>  result_tree_table_name, </div> |
| <div class="line"><a name="l02452"></a><span class="lineno"> 2452</span>  validation_table_name , </div> |
| <div class="line"><a name="l02453"></a><span class="lineno"> 2453</span>  num_classes</div> |
| <div class="line"><a name="l02454"></a><span class="lineno"> 2454</span>  );</div> |
| <div class="line"><a name="l02455"></a><span class="lineno"> 2455</span>  END IF;</div> |
| <div class="line"><a name="l02456"></a><span class="lineno"> 2456</span>  prune_time = clock_timestamp() - begin_prune;</div> |
| <div class="line"><a name="l02457"></a><span class="lineno"> 2457</span>  </div> |
| <div class="line"><a name="l02458"></a><span class="lineno"> 2458</span>  IF(verbosity > 0) THEN</div> |
| <div class="line"><a name="l02459"></a><span class="lineno"> 2459</span>  RAISE INFO 'time of sampling with replacement: %', bld_assoc_time;</div> |
| <div class="line"><a name="l02460"></a><span class="lineno"> 2460</span>  RAISE INFO 'time of finding best and calculating ACS: %', scv_acs_time;</div> |
| <div class="line"><a name="l02461"></a><span class="lineno"> 2461</span>  RAISE INFO 'time of calculating ACC: %', calc_acc_time;</div> |
| <div class="line"><a name="l02462"></a><span class="lineno"> 2462</span>  RAISE INFO 'time of Insert/update operation: %', ins_upd_time;</div> |
| <div class="line"><a name="l02463"></a><span class="lineno"> 2463</span>  RAISE INFO 'time of pruning: %', prune_time;</div> |
| <div class="line"><a name="l02464"></a><span class="lineno"> 2464</span>  RAISE INFO 'time of training: %', clock_timestamp() - begin_func_exec; </div> |
| <div class="line"><a name="l02465"></a><span class="lineno"> 2465</span>  END IF;</div> |
| <div class="line"><a name="l02466"></a><span class="lineno"> 2466</span>  </div> |
| <div class="line"><a name="l02467"></a><span class="lineno"> 2467</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02468"></a><span class="lineno"> 2468</span>  (</div> |
| <div class="line"><a name="l02469"></a><span class="lineno"> 2469</span>  'SELECT COUNT(id), max(array_upper(tree_location, 1))</div> |
| <div class="line"><a name="l02470"></a><span class="lineno"> 2470</span>  FROM %',</div> |
| <div class="line"><a name="l02471"></a><span class="lineno"> 2471</span>  ARRAY[</div> |
| <div class="line"><a name="l02472"></a><span class="lineno"> 2472</span>  result_tree_table_name</div> |
| <div class="line"><a name="l02473"></a><span class="lineno"> 2473</span>  ]</div> |
| <div class="line"><a name="l02474"></a><span class="lineno"> 2474</span>  ) INTO curstmt;</div> |
| <div class="line"><a name="l02475"></a><span class="lineno"> 2475</span>  </div> |
| <div class="line"><a name="l02476"></a><span class="lineno"> 2476</span>  EXECUTE curstmt INTO ret.num_tree_nodes, ret.max_tree_depth;</div> |
| <div class="line"><a name="l02477"></a><span class="lineno"> 2477</span>  </div> |
| <div class="line"><a name="l02478"></a><span class="lineno"> 2478</span>  ret.features_per_node = features_per_node; </div> |
| <div class="line"><a name="l02479"></a><span class="lineno"> 2479</span>  ret.num_of_samples = total_size;</div> |
| <div class="line"><a name="l02480"></a><span class="lineno"> 2480</span>  ret.calc_acc_time = calc_acc_time;</div> |
| <div class="line"><a name="l02481"></a><span class="lineno"> 2481</span>  ret.calc_pre_time = calc_pre_time;</div> |
| <div class="line"><a name="l02482"></a><span class="lineno"> 2482</span>  ret.update_time = ins_upd_time;</div> |
| <div class="line"><a name="l02483"></a><span class="lineno"> 2483</span>  ret.update_best = calc_update_best;</div> |
| <div class="line"><a name="l02484"></a><span class="lineno"> 2484</span>  ret.update_child = calc_update_child;</div> |
| <div class="line"><a name="l02485"></a><span class="lineno"> 2485</span>  ret.update_nid = calc_update_nid;</div> |
| <div class="line"><a name="l02486"></a><span class="lineno"> 2486</span>  ret.scv_acs_time = scv_acs_time; </div> |
| <div class="line"><a name="l02487"></a><span class="lineno"> 2487</span>  ret.prune_time = prune_time;</div> |
| <div class="line"><a name="l02488"></a><span class="lineno"> 2488</span>  </div> |
| <div class="line"><a name="l02489"></a><span class="lineno"> 2489</span>  RETURN ret;</div> |
| <div class="line"><a name="l02490"></a><span class="lineno"> 2490</span> END</div> |
| <div class="line"><a name="l02491"></a><span class="lineno"> 2491</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l02492"></a><span class="lineno"> 2492</span> </div> |
| <div class="line"><a name="l02493"></a><span class="lineno"> 2493</span> </div> |
| <div class="line"><a name="l02494"></a><span class="lineno"> 2494</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l02495"></a><span class="lineno"> 2495</span> <span class="comment"> * @brief This is an internal function for displaying one tree node in human </span></div> |
| <div class="line"><a name="l02496"></a><span class="lineno"> 2496</span> <span class="comment"> * readable format. It is the step function of aggregation named </span></div> |
| <div class="line"><a name="l02497"></a><span class="lineno"> 2497</span> <span class="comment"> * __display_tree_aggr.</span></div> |
| <div class="line"><a name="l02498"></a><span class="lineno"> 2498</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l02499"></a><span class="lineno"> 2499</span> <span class="comment"> * @param state This variable is used to store the accumulated tree </span></div> |
| <div class="line"><a name="l02500"></a><span class="lineno"> 2500</span> <span class="comment"> * display information.</span></div> |
| <div class="line"><a name="l02501"></a><span class="lineno"> 2501</span> <span class="comment"> * @param depth The depth of this node. </span></div> |
| <div class="line"><a name="l02502"></a><span class="lineno"> 2502</span> <span class="comment"> * @param is_cont Whether the feature used to split is continuous. </span></div> |
| <div class="line"><a name="l02503"></a><span class="lineno"> 2503</span> <span class="comment"> * @param feat_name The name of the feature used to split.</span></div> |
| <div class="line"><a name="l02504"></a><span class="lineno"> 2504</span> <span class="comment"> * @param curr_val The value of the splitting feature for this node.</span></div> |
| <div class="line"><a name="l02505"></a><span class="lineno"> 2505</span> <span class="comment"> * @param split_value For continuous feature, it specifies the split value. </span></div> |
| <div class="line"><a name="l02506"></a><span class="lineno"> 2506</span> <span class="comment"> * Otherwise, it is of no meaning.</span></div> |
| <div class="line"><a name="l02507"></a><span class="lineno"> 2507</span> <span class="comment"> * @param max_prob For those elements in this node, the probability that</span></div> |
| <div class="line"><a name="l02508"></a><span class="lineno"> 2508</span> <span class="comment"> * an element belongs to the max_class.</span></div> |
| <div class="line"><a name="l02509"></a><span class="lineno"> 2509</span> <span class="comment"> * @param max_class The class ID with the largest number of elements </span></div> |
| <div class="line"><a name="l02510"></a><span class="lineno"> 2510</span> <span class="comment"> * for those elements in this node.</span></div> |
| <div class="line"><a name="l02511"></a><span class="lineno"> 2511</span> <span class="comment"> * @param num_of_samples Total count of samples in this node. </span></div> |
| <div class="line"><a name="l02512"></a><span class="lineno"> 2512</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l02513"></a><span class="lineno"> 2513</span> <span class="comment"> * @return It returns the text containing the information of human </span></div> |
| <div class="line"><a name="l02514"></a><span class="lineno"> 2514</span> <span class="comment"> * readable information for trees.</span></div> |
| <div class="line"><a name="l02515"></a><span class="lineno"> 2515</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l02516"></a><span class="lineno"> 2516</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l02517"></a><span class="lineno"> 2517</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__display_node_sfunc</div> |
| <div class="line"><a name="l02518"></a><span class="lineno"> 2518</span>  (</div> |
| <div class="line"><a name="l02519"></a><span class="lineno"> 2519</span>  state TEXT,</div> |
| <div class="line"><a name="l02520"></a><span class="lineno"> 2520</span>  depth INT,</div> |
| <div class="line"><a name="l02521"></a><span class="lineno"> 2521</span>  is_cont BOOLEAN,</div> |
| <div class="line"><a name="l02522"></a><span class="lineno"> 2522</span>  feat_name TEXT,</div> |
| <div class="line"><a name="l02523"></a><span class="lineno"> 2523</span>  curr_val TEXT,</div> |
| <div class="line"><a name="l02524"></a><span class="lineno"> 2524</span>  split_value FLOAT8,</div> |
| <div class="line"><a name="l02525"></a><span class="lineno"> 2525</span>  max_prob FLOAT8,</div> |
| <div class="line"><a name="l02526"></a><span class="lineno"> 2526</span>  max_class TEXT,</div> |
| <div class="line"><a name="l02527"></a><span class="lineno"> 2527</span>  num_of_samples INT</div> |
| <div class="line"><a name="l02528"></a><span class="lineno"> 2528</span>  ) </div> |
| <div class="line"><a name="l02529"></a><span class="lineno"> 2529</span> RETURNS TEXT AS $$ </div> |
| <div class="line"><a name="l02530"></a><span class="lineno"> 2530</span> DECLARE</div> |
| <div class="line"><a name="l02531"></a><span class="lineno"> 2531</span>  ret TEXT := '';</div> |
| <div class="line"><a name="l02532"></a><span class="lineno"> 2532</span>  index INT;</div> |
| <div class="line"><a name="l02533"></a><span class="lineno"> 2533</span> BEGIN</div> |
| <div class="line"><a name="l02534"></a><span class="lineno"> 2534</span>  -- We add indentation based on the depth.</div> |
| <div class="line"><a name="l02535"></a><span class="lineno"> 2535</span>  FOR index IN 0..depth LOOP</div> |
| <div class="line"><a name="l02536"></a><span class="lineno"> 2536</span>  ret = ret || ' ';</div> |
| <div class="line"><a name="l02537"></a><span class="lineno"> 2537</span>  END LOOP;</div> |
| <div class="line"><a name="l02538"></a><span class="lineno"> 2538</span>  </div> |
| <div class="line"><a name="l02539"></a><span class="lineno"> 2539</span>  IF (depth > 0) THEN</div> |
| <div class="line"><a name="l02540"></a><span class="lineno"> 2540</span>  ret = ret ||coalesce(feat_name,'null')||': ';</div> |
| <div class="line"><a name="l02541"></a><span class="lineno"> 2541</span>  -- For continuous features, there are two splits.</div> |
| <div class="line"><a name="l02542"></a><span class="lineno"> 2542</span>  -- We will mark curr_val to 1 for '<='. Otherwise, </div> |
| <div class="line"><a name="l02543"></a><span class="lineno"> 2543</span>  -- we will mark curr_val to 2.</div> |
| <div class="line"><a name="l02544"></a><span class="lineno"> 2544</span>  IF (is_cont) THEN</div> |
| <div class="line"><a name="l02545"></a><span class="lineno"> 2545</span>  IF (curr_val::INT = 1) THEN</div> |
| <div class="line"><a name="l02546"></a><span class="lineno"> 2546</span>  ret = ret || ' <= ';</div> |
| <div class="line"><a name="l02547"></a><span class="lineno"> 2547</span>  ELSE</div> |
| <div class="line"><a name="l02548"></a><span class="lineno"> 2548</span>  ret = ret || ' > ';</div> |
| <div class="line"><a name="l02549"></a><span class="lineno"> 2549</span>  END IF;</div> |
| <div class="line"><a name="l02550"></a><span class="lineno"> 2550</span>  ret = ret||coalesce(split_value,0)||' ';</div> |
| <div class="line"><a name="l02551"></a><span class="lineno"> 2551</span>  ELSE</div> |
| <div class="line"><a name="l02552"></a><span class="lineno"> 2552</span>  ret = ret||' = '||coalesce(curr_val,'null')||' ';</div> |
| <div class="line"><a name="l02553"></a><span class="lineno"> 2553</span>  END IF;</div> |
| <div class="line"><a name="l02554"></a><span class="lineno"> 2554</span>  ELSE</div> |
| <div class="line"><a name="l02555"></a><span class="lineno"> 2555</span>  ret = ret||'Root Node ';</div> |
| <div class="line"><a name="l02556"></a><span class="lineno"> 2556</span>  END IF;</div> |
| <div class="line"><a name="l02557"></a><span class="lineno"> 2557</span> </div> |
| <div class="line"><a name="l02558"></a><span class="lineno"> 2558</span>  ret = ret || </div> |
| <div class="line"><a name="l02559"></a><span class="lineno"> 2559</span>  ' : class(' || </div> |
| <div class="line"><a name="l02560"></a><span class="lineno"> 2560</span>  coalesce(max_class,null) || </div> |
| <div class="line"><a name="l02561"></a><span class="lineno"> 2561</span>  ') num_elements(' || </div> |
| <div class="line"><a name="l02562"></a><span class="lineno"> 2562</span>  coalesce(num_of_samples,0) || </div> |
| <div class="line"><a name="l02563"></a><span class="lineno"> 2563</span>  ') predict_prob(' ||</div> |
| <div class="line"><a name="l02564"></a><span class="lineno"> 2564</span>  coalesce(max_prob,0) ||</div> |
| <div class="line"><a name="l02565"></a><span class="lineno"> 2565</span>  ')';</div> |
| <div class="line"><a name="l02566"></a><span class="lineno"> 2566</span> </div> |
| <div class="line"><a name="l02567"></a><span class="lineno"> 2567</span>  ret = ret || E'\n';</div> |
| <div class="line"><a name="l02568"></a><span class="lineno"> 2568</span> </div> |
| <div class="line"><a name="l02569"></a><span class="lineno"> 2569</span>  -- If there exists information, append the information</div> |
| <div class="line"><a name="l02570"></a><span class="lineno"> 2570</span>  -- for this node.</div> |
| <div class="line"><a name="l02571"></a><span class="lineno"> 2571</span>  IF (state IS NOT NULL) THEN</div> |
| <div class="line"><a name="l02572"></a><span class="lineno"> 2572</span>  ret = state || ret;</div> |
| <div class="line"><a name="l02573"></a><span class="lineno"> 2573</span>  END IF;</div> |
| <div class="line"><a name="l02574"></a><span class="lineno"> 2574</span>  </div> |
| <div class="line"><a name="l02575"></a><span class="lineno"> 2575</span>  RETURN ret;</div> |
| <div class="line"><a name="l02576"></a><span class="lineno"> 2576</span> END </div> |
| <div class="line"><a name="l02577"></a><span class="lineno"> 2577</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l02578"></a><span class="lineno"> 2578</span> </div> |
| <div class="line"><a name="l02579"></a><span class="lineno"> 2579</span> </div> |
| <div class="line"><a name="l02580"></a><span class="lineno"> 2580</span> DROP AGGREGATE IF EXISTS MADLIB_SCHEMA.__display_tree_aggr</div> |
| <div class="line"><a name="l02581"></a><span class="lineno"> 2581</span>  (</div> |
| <div class="line"><a name="l02582"></a><span class="lineno"> 2582</span>  INT, -- depth</div> |
| <div class="line"><a name="l02583"></a><span class="lineno"> 2583</span>  BOOLEAN, -- is_cont</div> |
| <div class="line"><a name="l02584"></a><span class="lineno"> 2584</span>  TEXT, -- feature name</div> |
| <div class="line"><a name="l02585"></a><span class="lineno"> 2585</span>  TEXT, -- curr_val</div> |
| <div class="line"><a name="l02586"></a><span class="lineno"> 2586</span>  FLOAT8, -- split value</div> |
| <div class="line"><a name="l02587"></a><span class="lineno"> 2587</span>  FLOAT8, -- max_probability</div> |
| <div class="line"><a name="l02588"></a><span class="lineno"> 2588</span>  TEXT, -- max_class</div> |
| <div class="line"><a name="l02589"></a><span class="lineno"> 2589</span>  INT -- num_of_samples</div> |
| <div class="line"><a name="l02590"></a><span class="lineno"> 2590</span>  ) CASCADE;</div> |
| <div class="line"><a name="l02591"></a><span class="lineno"> 2591</span> CREATE </div> |
| <div class="line"><a name="l02592"></a><span class="lineno"> 2592</span> m4_ifdef(`__GREENPLUM__', m4_ifdef(`__HAS_ORDERED_AGGREGATES__', `ORDERED'))</div> |
| <div class="line"><a name="l02593"></a><span class="lineno"> 2593</span> AGGREGATE MADLIB_SCHEMA.__display_tree_aggr</div> |
| <div class="line"><a name="l02594"></a><span class="lineno"> 2594</span>  (</div> |
| <div class="line"><a name="l02595"></a><span class="lineno"> 2595</span>  INT, -- depth</div> |
| <div class="line"><a name="l02596"></a><span class="lineno"> 2596</span>  BOOLEAN, -- is_cont</div> |
| <div class="line"><a name="l02597"></a><span class="lineno"> 2597</span>  TEXT, -- feature name</div> |
| <div class="line"><a name="l02598"></a><span class="lineno"> 2598</span>  TEXT, -- curr_val</div> |
| <div class="line"><a name="l02599"></a><span class="lineno"> 2599</span>  FLOAT8, -- split value</div> |
| <div class="line"><a name="l02600"></a><span class="lineno"> 2600</span>  FLOAT8, -- max_probability</div> |
| <div class="line"><a name="l02601"></a><span class="lineno"> 2601</span>  TEXT, -- max_class</div> |
| <div class="line"><a name="l02602"></a><span class="lineno"> 2602</span>  INT -- num_of_samples</div> |
| <div class="line"><a name="l02603"></a><span class="lineno"> 2603</span>  ) </div> |
| <div class="line"><a name="l02604"></a><span class="lineno"> 2604</span> (</div> |
| <div class="line"><a name="l02605"></a><span class="lineno"> 2605</span>  SFUNC=MADLIB_SCHEMA.__display_node_sfunc,</div> |
| <div class="line"><a name="l02606"></a><span class="lineno"> 2606</span>  STYPE=TEXT</div> |
| <div class="line"><a name="l02607"></a><span class="lineno"> 2607</span> );</div> |
| <div class="line"><a name="l02608"></a><span class="lineno"> 2608</span> </div> |
| <div class="line"><a name="l02609"></a><span class="lineno"> 2609</span> </div> |
| <div class="line"><a name="l02610"></a><span class="lineno"> 2610</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l02611"></a><span class="lineno"> 2611</span> <span class="comment"> * @brief Display the trained model with human readable format. This function</span></div> |
| <div class="line"><a name="l02612"></a><span class="lineno"> 2612</span> <span class="comment"> * leverages ordered aggregate to display the tree with only one scan of</span></div> |
| <div class="line"><a name="l02613"></a><span class="lineno"> 2613</span> <span class="comment"> * the tree_table.</span></div> |
| <div class="line"><a name="l02614"></a><span class="lineno"> 2614</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l02615"></a><span class="lineno"> 2615</span> <span class="comment"> * @param tree_table The full name of the tree table. </span></div> |
| <div class="line"><a name="l02616"></a><span class="lineno"> 2616</span> <span class="comment"> * @param tree_id The array contains the IDs of the trees to be displayed.</span></div> |
| <div class="line"><a name="l02617"></a><span class="lineno"> 2617</span> <span class="comment"> * @param max_depth The max depth to be displayed. If it is set to null,</span></div> |
| <div class="line"><a name="l02618"></a><span class="lineno"> 2618</span> <span class="comment"> * this function will show all levels. </span></div> |
| <div class="line"><a name="l02619"></a><span class="lineno"> 2619</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l02620"></a><span class="lineno"> 2620</span> <span class="comment"> * @return The text representing the tree with human readable format.</span></div> |
| <div class="line"><a name="l02621"></a><span class="lineno"> 2621</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l02622"></a><span class="lineno"> 2622</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l02623"></a><span class="lineno"> 2623</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__treemodel_display_with_ordered_aggr</div> |
| <div class="line"><a name="l02624"></a><span class="lineno"> 2624</span>  (</div> |
| <div class="line"><a name="l02625"></a><span class="lineno"> 2625</span>  tree_table TEXT,</div> |
| <div class="line"><a name="l02626"></a><span class="lineno"> 2626</span>  tree_id INT[], </div> |
| <div class="line"><a name="l02627"></a><span class="lineno"> 2627</span>  max_depth INT</div> |
| <div class="line"><a name="l02628"></a><span class="lineno"> 2628</span>  ) </div> |
| <div class="line"><a name="l02629"></a><span class="lineno"> 2629</span> RETURNS SETOF TEXT AS $$</div> |
| <div class="line"><a name="l02630"></a><span class="lineno"> 2630</span> DECLARE</div> |
| <div class="line"><a name="l02631"></a><span class="lineno"> 2631</span>  metatable_name TEXT := null;</div> |
| <div class="line"><a name="l02632"></a><span class="lineno"> 2632</span>  curr_stmt TEXT := null;</div> |
| <div class="line"><a name="l02633"></a><span class="lineno"> 2633</span>  feature_name TEXT := null;</div> |
| <div class="line"><a name="l02634"></a><span class="lineno"> 2634</span>  table_name TEXT := null;</div> |
| <div class="line"><a name="l02635"></a><span class="lineno"> 2635</span>  result TEXT := '';</div> |
| <div class="line"><a name="l02636"></a><span class="lineno"> 2636</span>  result_rec RECORD;</div> |
| <div class="line"><a name="l02637"></a><span class="lineno"> 2637</span> BEGIN</div> |
| <div class="line"><a name="l02638"></a><span class="lineno"> 2638</span>  PERFORM MADLIB_SCHEMA.__assert_table</div> |
| <div class="line"><a name="l02639"></a><span class="lineno"> 2639</span>  (</div> |
| <div class="line"><a name="l02640"></a><span class="lineno"> 2640</span>  tree_table,</div> |
| <div class="line"><a name="l02641"></a><span class="lineno"> 2641</span>  't'</div> |
| <div class="line"><a name="l02642"></a><span class="lineno"> 2642</span>  ); </div> |
| <div class="line"><a name="l02643"></a><span class="lineno"> 2643</span>  </div> |
| <div class="line"><a name="l02644"></a><span class="lineno"> 2644</span>  metatable_name = MADLIB_SCHEMA.__get_metatable_name( tree_table );</div> |
| <div class="line"><a name="l02645"></a><span class="lineno"> 2645</span>  </div> |
| <div class="line"><a name="l02646"></a><span class="lineno"> 2646</span>  -- This table is used for tree display.</div> |
| <div class="line"><a name="l02647"></a><span class="lineno"> 2647</span>  -- It is filled with the original information before</div> |
| <div class="line"><a name="l02648"></a><span class="lineno"> 2648</span>  -- encoding to facilitate the display procedure.</div> |
| <div class="line"><a name="l02649"></a><span class="lineno"> 2649</span>  DROP TABLE IF EXISTS auxiliary_tree_display;</div> |
| <div class="line"><a name="l02650"></a><span class="lineno"> 2650</span>  CREATE TEMP TABLE auxiliary_tree_display</div> |
| <div class="line"><a name="l02651"></a><span class="lineno"> 2651</span>  (</div> |
| <div class="line"><a name="l02652"></a><span class="lineno"> 2652</span>  tid INT,</div> |
| <div class="line"><a name="l02653"></a><span class="lineno"> 2653</span>  id INT,</div> |
| <div class="line"><a name="l02654"></a><span class="lineno"> 2654</span>  tree_location INT[],</div> |
| <div class="line"><a name="l02655"></a><span class="lineno"> 2655</span>  probability FLOAT8, </div> |
| <div class="line"><a name="l02656"></a><span class="lineno"> 2656</span>  max_class TEXT,</div> |
| <div class="line"><a name="l02657"></a><span class="lineno"> 2657</span>  num_of_samples INT,</div> |
| <div class="line"><a name="l02658"></a><span class="lineno"> 2658</span>  parent_id INT,</div> |
| <div class="line"><a name="l02659"></a><span class="lineno"> 2659</span>  curr_value TEXT,</div> |
| <div class="line"><a name="l02660"></a><span class="lineno"> 2660</span>  parent_feature_id INT,</div> |
| <div class="line"><a name="l02661"></a><span class="lineno"> 2661</span>  is_parent_feature_cont BOOLEAN,</div> |
| <div class="line"><a name="l02662"></a><span class="lineno"> 2662</span>  parent_split_value FLOAT8,</div> |
| <div class="line"><a name="l02663"></a><span class="lineno"> 2663</span>  parent_feature_name TEXT</div> |
| <div class="line"><a name="l02664"></a><span class="lineno"> 2664</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (id)');</div> |
| <div class="line"><a name="l02665"></a><span class="lineno"> 2665</span>  </div> |
| <div class="line"><a name="l02666"></a><span class="lineno"> 2666</span>  -- We made a self join for the tree table. For each node, we get the </div> |
| <div class="line"><a name="l02667"></a><span class="lineno"> 2667</span>  -- feature information at its parent node so as to display this node. </div> |
| <div class="line"><a name="l02668"></a><span class="lineno"> 2668</span>  SELECT MADLIB_SCHEMA.__format(</div> |
| <div class="line"><a name="l02669"></a><span class="lineno"> 2669</span>  'INSERT INTO auxiliary_tree_display SELECT m.*, </div> |
| <div class="line"><a name="l02670"></a><span class="lineno"> 2670</span>  n.column_name as parent_feature_name</div> |
| <div class="line"><a name="l02671"></a><span class="lineno"> 2671</span>  FROM </div> |
| <div class="line"><a name="l02672"></a><span class="lineno"> 2672</span>  (SELECT * FROM</div> |
| <div class="line"><a name="l02673"></a><span class="lineno"> 2673</span>  (SELECT t1.tid,t1.id, t1.tree_location,</div> |
| <div class="line"><a name="l02674"></a><span class="lineno"> 2674</span>  t1.probability,t1.max_class::TEXT,</div> |
| <div class="line"><a name="l02675"></a><span class="lineno"> 2675</span>  t1.num_of_samples,t1.parent_id, </div> |
| <div class="line"><a name="l02676"></a><span class="lineno"> 2676</span>  t1.tree_location[array_upper(t1.tree_location,1)]::TEXT </div> |
| <div class="line"><a name="l02677"></a><span class="lineno"> 2677</span>  as curr_value, </div> |
| <div class="line"><a name="l02678"></a><span class="lineno"> 2678</span>  t2.feature as parent_feature_id, </div> |
| <div class="line"><a name="l02679"></a><span class="lineno"> 2679</span>  t2.is_cont as is_parent_feature_cont, </div> |
| <div class="line"><a name="l02680"></a><span class="lineno"> 2680</span>  t2.split_value as parent_split_value </div> |
| <div class="line"><a name="l02681"></a><span class="lineno"> 2681</span>  FROM % t1 LEFT JOIN % t2 ON </div> |
| <div class="line"><a name="l02682"></a><span class="lineno"> 2682</span>  (t1.parent_id = t2.id AND </div> |
| <div class="line"><a name="l02683"></a><span class="lineno"> 2683</span>  (coalesce(t1.tid,0)=coalesce(t2.tid,0)) ) ) l</div> |
| <div class="line"><a name="l02684"></a><span class="lineno"> 2684</span>  WHERE l.tid in ( % ) ) m </div> |
| <div class="line"><a name="l02685"></a><span class="lineno"> 2685</span>  LEFT JOIN % n </div> |
| <div class="line"><a name="l02686"></a><span class="lineno"> 2686</span>  on m.parent_feature_id = n.id;',</div> |
| <div class="line"><a name="l02687"></a><span class="lineno"> 2687</span>  ARRAY[</div> |
| <div class="line"><a name="l02688"></a><span class="lineno"> 2688</span>  tree_table,</div> |
| <div class="line"><a name="l02689"></a><span class="lineno"> 2689</span>  tree_table,</div> |
| <div class="line"><a name="l02690"></a><span class="lineno"> 2690</span>  array_to_string(tree_id,','),</div> |
| <div class="line"><a name="l02691"></a><span class="lineno"> 2691</span>  metatable_name</div> |
| <div class="line"><a name="l02692"></a><span class="lineno"> 2692</span>  ]</div> |
| <div class="line"><a name="l02693"></a><span class="lineno"> 2693</span>  )</div> |
| <div class="line"><a name="l02694"></a><span class="lineno"> 2694</span>  INTO curr_stmt; </div> |
| <div class="line"><a name="l02695"></a><span class="lineno"> 2695</span>  EXECUTE curr_stmt;</div> |
| <div class="line"><a name="l02696"></a><span class="lineno"> 2696</span> </div> |
| <div class="line"><a name="l02697"></a><span class="lineno"> 2697</span>  -- Get the metatable storing the encoding information of class.</div> |
| <div class="line"><a name="l02698"></a><span class="lineno"> 2698</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02699"></a><span class="lineno"> 2699</span>  (</div> |
| <div class="line"><a name="l02700"></a><span class="lineno"> 2700</span>  'SELECT </div> |
| <div class="line"><a name="l02701"></a><span class="lineno"> 2701</span>  column_name,</div> |
| <div class="line"><a name="l02702"></a><span class="lineno"> 2702</span>  MADLIB_SCHEMA.__regclass_to_text(table_oid) as table_name</div> |
| <div class="line"><a name="l02703"></a><span class="lineno"> 2703</span>  FROM %</div> |
| <div class="line"><a name="l02704"></a><span class="lineno"> 2704</span>  WHERE column_type=''c'' LIMIT 1',</div> |
| <div class="line"><a name="l02705"></a><span class="lineno"> 2705</span>  ARRAY[</div> |
| <div class="line"><a name="l02706"></a><span class="lineno"> 2706</span>  metatable_name</div> |
| <div class="line"><a name="l02707"></a><span class="lineno"> 2707</span>  ]</div> |
| <div class="line"><a name="l02708"></a><span class="lineno"> 2708</span>  ) INTO curr_stmt;</div> |
| <div class="line"><a name="l02709"></a><span class="lineno"> 2709</span>  </div> |
| <div class="line"><a name="l02710"></a><span class="lineno"> 2710</span>  EXECUTE curr_stmt INTO result_rec;</div> |
| <div class="line"><a name="l02711"></a><span class="lineno"> 2711</span>  </div> |
| <div class="line"><a name="l02712"></a><span class="lineno"> 2712</span>  table_name = result_rec.table_name;</div> |
| <div class="line"><a name="l02713"></a><span class="lineno"> 2713</span>  </div> |
| <div class="line"><a name="l02714"></a><span class="lineno"> 2714</span>  IF (table_name IS NOT NULL) THEN</div> |
| <div class="line"><a name="l02715"></a><span class="lineno"> 2715</span>  -- Convert back for the class column.</div> |
| <div class="line"><a name="l02716"></a><span class="lineno"> 2716</span>  SELECT MADLIB_SCHEMA.__format(</div> |
| <div class="line"><a name="l02717"></a><span class="lineno"> 2717</span>  'UPDATE auxiliary_tree_display n </div> |
| <div class="line"><a name="l02718"></a><span class="lineno"> 2718</span>  SET max_class = MADLIB_SCHEMA.__to_char(m.fval) </div> |
| <div class="line"><a name="l02719"></a><span class="lineno"> 2719</span>  FROM % m </div> |
| <div class="line"><a name="l02720"></a><span class="lineno"> 2720</span>  WHERE m.code = n.max_class::INT</div> |
| <div class="line"><a name="l02721"></a><span class="lineno"> 2721</span>  ',</div> |
| <div class="line"><a name="l02722"></a><span class="lineno"> 2722</span>  ARRAY[</div> |
| <div class="line"><a name="l02723"></a><span class="lineno"> 2723</span>  table_name</div> |
| <div class="line"><a name="l02724"></a><span class="lineno"> 2724</span>  ]</div> |
| <div class="line"><a name="l02725"></a><span class="lineno"> 2725</span>  )</div> |
| <div class="line"><a name="l02726"></a><span class="lineno"> 2726</span>  INTO curr_stmt; </div> |
| <div class="line"><a name="l02727"></a><span class="lineno"> 2727</span>  EXECUTE curr_stmt;</div> |
| <div class="line"><a name="l02728"></a><span class="lineno"> 2728</span>  END IF;</div> |
| <div class="line"><a name="l02729"></a><span class="lineno"> 2729</span> </div> |
| <div class="line"><a name="l02730"></a><span class="lineno"> 2730</span>  -- Get the metatables storing the encoding information for discrete features.</div> |
| <div class="line"><a name="l02731"></a><span class="lineno"> 2731</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02732"></a><span class="lineno"> 2732</span>  (</div> |
| <div class="line"><a name="l02733"></a><span class="lineno"> 2733</span>  'SELECT </div> |
| <div class="line"><a name="l02734"></a><span class="lineno"> 2734</span>  id,</div> |
| <div class="line"><a name="l02735"></a><span class="lineno"> 2735</span>  column_name,</div> |
| <div class="line"><a name="l02736"></a><span class="lineno"> 2736</span>  MADLIB_SCHEMA.__regclass_to_text(table_oid) as table_name </div> |
| <div class="line"><a name="l02737"></a><span class="lineno"> 2737</span>  FROM % </div> |
| <div class="line"><a name="l02738"></a><span class="lineno"> 2738</span>  WHERE NOT is_cont AND column_type=''f'';',</div> |
| <div class="line"><a name="l02739"></a><span class="lineno"> 2739</span>  ARRAY[</div> |
| <div class="line"><a name="l02740"></a><span class="lineno"> 2740</span>  metatable_name</div> |
| <div class="line"><a name="l02741"></a><span class="lineno"> 2741</span>  ]</div> |
| <div class="line"><a name="l02742"></a><span class="lineno"> 2742</span>  )</div> |
| <div class="line"><a name="l02743"></a><span class="lineno"> 2743</span>  INTO curr_stmt; </div> |
| <div class="line"><a name="l02744"></a><span class="lineno"> 2744</span>  </div> |
| <div class="line"><a name="l02745"></a><span class="lineno"> 2745</span>  -- Convert back for discrete features.</div> |
| <div class="line"><a name="l02746"></a><span class="lineno"> 2746</span>  FOR result_rec IN EXECUTE (curr_stmt) LOOP</div> |
| <div class="line"><a name="l02747"></a><span class="lineno"> 2747</span>  SELECT MADLIB_SCHEMA.__format(</div> |
| <div class="line"><a name="l02748"></a><span class="lineno"> 2748</span>  'UPDATE auxiliary_tree_display n </div> |
| <div class="line"><a name="l02749"></a><span class="lineno"> 2749</span>  SET curr_value = MADLIB_SCHEMA.__to_char(m.fval) </div> |
| <div class="line"><a name="l02750"></a><span class="lineno"> 2750</span>  FROM % m </div> |
| <div class="line"><a name="l02751"></a><span class="lineno"> 2751</span>  WHERE m.code::INT = n.curr_value::INT AND</div> |
| <div class="line"><a name="l02752"></a><span class="lineno"> 2752</span>  m.fid = % AND </div> |
| <div class="line"><a name="l02753"></a><span class="lineno"> 2753</span>  n.parent_feature_name = %</div> |
| <div class="line"><a name="l02754"></a><span class="lineno"> 2754</span>  ',</div> |
| <div class="line"><a name="l02755"></a><span class="lineno"> 2755</span>  ARRAY[</div> |
| <div class="line"><a name="l02756"></a><span class="lineno"> 2756</span>  result_rec.table_name,</div> |
| <div class="line"><a name="l02757"></a><span class="lineno"> 2757</span>  result_rec.id::TEXT,</div> |
| <div class="line"><a name="l02758"></a><span class="lineno"> 2758</span>  quote_literal(result_rec.column_name)</div> |
| <div class="line"><a name="l02759"></a><span class="lineno"> 2759</span>  ]</div> |
| <div class="line"><a name="l02760"></a><span class="lineno"> 2760</span>  )</div> |
| <div class="line"><a name="l02761"></a><span class="lineno"> 2761</span>  INTO curr_stmt; </div> |
| <div class="line"><a name="l02762"></a><span class="lineno"> 2762</span>  EXECUTE curr_stmt; </div> |
| <div class="line"><a name="l02763"></a><span class="lineno"> 2763</span>  END LOOP;</div> |
| <div class="line"><a name="l02764"></a><span class="lineno"> 2764</span> </div> |
| <div class="line"><a name="l02765"></a><span class="lineno"> 2765</span>  -- Now we already get all the information. Invoke the</div> |
| <div class="line"><a name="l02766"></a><span class="lineno"> 2766</span>  -- aggregation to show the tree.</div> |
| <div class="line"><a name="l02767"></a><span class="lineno"> 2767</span>  -- If we order by tree_location, we can get the sequence </div> |
| <div class="line"><a name="l02768"></a><span class="lineno"> 2768</span>  -- of depth first traversal.</div> |
| <div class="line"><a name="l02769"></a><span class="lineno"> 2769</span>  curr_stmt = 'SELECT tid,MADLIB_SCHEMA.__display_tree_aggr(</div> |
| <div class="line"><a name="l02770"></a><span class="lineno"> 2770</span>  array_upper(tree_location,1)-1,</div> |
| <div class="line"><a name="l02771"></a><span class="lineno"> 2771</span>  is_parent_feature_cont,</div> |
| <div class="line"><a name="l02772"></a><span class="lineno"> 2772</span>  parent_feature_name,</div> |
| <div class="line"><a name="l02773"></a><span class="lineno"> 2773</span>  curr_value,</div> |
| <div class="line"><a name="l02774"></a><span class="lineno"> 2774</span>  parent_split_value,</div> |
| <div class="line"><a name="l02775"></a><span class="lineno"> 2775</span>  probability,</div> |
| <div class="line"><a name="l02776"></a><span class="lineno"> 2776</span>  max_class,</div> |
| <div class="line"><a name="l02777"></a><span class="lineno"> 2777</span>  num_of_samples </div> |
| <div class="line"><a name="l02778"></a><span class="lineno"> 2778</span>  order by tree_location) AS disp_str</div> |
| <div class="line"><a name="l02779"></a><span class="lineno"> 2779</span>  FROM auxiliary_tree_display';</div> |
| <div class="line"><a name="l02780"></a><span class="lineno"> 2780</span> </div> |
| <div class="line"><a name="l02781"></a><span class="lineno"> 2781</span>  IF (max_depth IS NOT NULL) THEN</div> |
| <div class="line"><a name="l02782"></a><span class="lineno"> 2782</span>  curr_stmt = curr_stmt ||</div> |
| <div class="line"><a name="l02783"></a><span class="lineno"> 2783</span>  ' WHERE array_upper(tree_location,1) - 1 <=' ||</div> |
| <div class="line"><a name="l02784"></a><span class="lineno"> 2784</span>  max_depth;</div> |
| <div class="line"><a name="l02785"></a><span class="lineno"> 2785</span>  END IF;</div> |
| <div class="line"><a name="l02786"></a><span class="lineno"> 2786</span>  </div> |
| <div class="line"><a name="l02787"></a><span class="lineno"> 2787</span>  curr_stmt = curr_stmt||' GROUP BY tid ORDER BY tid;';</div> |
| <div class="line"><a name="l02788"></a><span class="lineno"> 2788</span>  </div> |
| <div class="line"><a name="l02789"></a><span class="lineno"> 2789</span>  FOR result_rec IN EXECUTE curr_stmt LOOP</div> |
| <div class="line"><a name="l02790"></a><span class="lineno"> 2790</span>  SELECT MADLIB_SCHEMA.__format(</div> |
| <div class="line"><a name="l02791"></a><span class="lineno"> 2791</span>  E'\nTree %\n%',</div> |
| <div class="line"><a name="l02792"></a><span class="lineno"> 2792</span>  ARRAY[</div> |
| <div class="line"><a name="l02793"></a><span class="lineno"> 2793</span>  result_rec.tid::TEXT,</div> |
| <div class="line"><a name="l02794"></a><span class="lineno"> 2794</span>  result_rec.disp_str</div> |
| <div class="line"><a name="l02795"></a><span class="lineno"> 2795</span>  ]</div> |
| <div class="line"><a name="l02796"></a><span class="lineno"> 2796</span>  )</div> |
| <div class="line"><a name="l02797"></a><span class="lineno"> 2797</span>  INTO result; </div> |
| <div class="line"><a name="l02798"></a><span class="lineno"> 2798</span>  RETURN NEXT result;</div> |
| <div class="line"><a name="l02799"></a><span class="lineno"> 2799</span>  --RETURN NEXT E'\nTree '||result_rec.tid||E'\n'||result_rec.disp_str;</div> |
| <div class="line"><a name="l02800"></a><span class="lineno"> 2800</span>  END LOOP;</div> |
| <div class="line"><a name="l02801"></a><span class="lineno"> 2801</span>  RETURN;</div> |
| <div class="line"><a name="l02802"></a><span class="lineno"> 2802</span> END $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l02803"></a><span class="lineno"> 2803</span> </div> |
| <div class="line"><a name="l02804"></a><span class="lineno"> 2804</span> </div> |
| <div class="line"><a name="l02805"></a><span class="lineno"> 2805</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l02806"></a><span class="lineno"> 2806</span> <span class="comment"> * @brief This is an internal function for displaying the tree in human readable</span></div> |
| <div class="line"><a name="l02807"></a><span class="lineno"> 2807</span> <span class="comment"> * format. It uses the depth-first strategy to traverse a tree and print </span></div> |
| <div class="line"><a name="l02808"></a><span class="lineno"> 2808</span> <span class="comment"> * values. This function is used on databases, e.g. PG 8.4, that do not </span></div> |
| <div class="line"><a name="l02809"></a><span class="lineno"> 2809</span> <span class="comment"> * support ordered aggregate.</span></div> |
| <div class="line"><a name="l02810"></a><span class="lineno"> 2810</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l02811"></a><span class="lineno"> 2811</span> <span class="comment"> * @param tree_table The full name of the tree table. </span></div> |
| <div class="line"><a name="l02812"></a><span class="lineno"> 2812</span> <span class="comment"> * @param id The ID of current node. This node and all of its </span></div> |
| <div class="line"><a name="l02813"></a><span class="lineno"> 2813</span> <span class="comment"> * children are displayed.</span></div> |
| <div class="line"><a name="l02814"></a><span class="lineno"> 2814</span> <span class="comment"> * @param feature_id The ID of a feature, which is used to split in the </span></div> |
| <div class="line"><a name="l02815"></a><span class="lineno"> 2815</span> <span class="comment"> * parent of current node.</span></div> |
| <div class="line"><a name="l02816"></a><span class="lineno"> 2816</span> <span class="comment"> * @param depth The depth of current node.</span></div> |
| <div class="line"><a name="l02817"></a><span class="lineno"> 2817</span> <span class="comment"> * @param is_cont It specifies whether the feature denoted by 'feature_id'</span></div> |
| <div class="line"><a name="l02818"></a><span class="lineno"> 2818</span> <span class="comment"> * is continuous or not.</span></div> |
| <div class="line"><a name="l02819"></a><span class="lineno"> 2819</span> <span class="comment"> * @param split_value For continuous feature, it specifies the split value. </span></div> |
| <div class="line"><a name="l02820"></a><span class="lineno"> 2820</span> <span class="comment"> * Otherwise, it is of no meaning.</span></div> |
| <div class="line"><a name="l02821"></a><span class="lineno"> 2821</span> <span class="comment"> * @param metatable_name For tabular format, this table contains the meta data</span></div> |
| <div class="line"><a name="l02822"></a><span class="lineno"> 2822</span> <span class="comment"> * to encode the input table.</span></div> |
| <div class="line"><a name="l02823"></a><span class="lineno"> 2823</span> <span class="comment"> * @param max_depth The max depth to be displayed. If it is set to null,</span></div> |
| <div class="line"><a name="l02824"></a><span class="lineno"> 2824</span> <span class="comment"> * this function will show all levels. </span></div> |
| <div class="line"><a name="l02825"></a><span class="lineno"> 2825</span> <span class="comment"> * @param tree_id The ID of the tree to be displayed.</span></div> |
| <div class="line"><a name="l02826"></a><span class="lineno"> 2826</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l02827"></a><span class="lineno"> 2827</span> <span class="comment"> * @return The text representing the tree with human readable format.</span></div> |
| <div class="line"><a name="l02828"></a><span class="lineno"> 2828</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l02829"></a><span class="lineno"> 2829</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l02830"></a><span class="lineno"> 2830</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__display_tree_no_ordered_aggr</div> |
| <div class="line"><a name="l02831"></a><span class="lineno"> 2831</span>  (</div> |
| <div class="line"><a name="l02832"></a><span class="lineno"> 2832</span>  tree_table TEXT, </div> |
| <div class="line"><a name="l02833"></a><span class="lineno"> 2833</span>  id INT, </div> |
| <div class="line"><a name="l02834"></a><span class="lineno"> 2834</span>  feature_id INT, </div> |
| <div class="line"><a name="l02835"></a><span class="lineno"> 2835</span>  depth INT, </div> |
| <div class="line"><a name="l02836"></a><span class="lineno"> 2836</span>  is_cont BOOLEAN, </div> |
| <div class="line"><a name="l02837"></a><span class="lineno"> 2837</span>  split_value FLOAT,</div> |
| <div class="line"><a name="l02838"></a><span class="lineno"> 2838</span>  metatable_name TEXT,</div> |
| <div class="line"><a name="l02839"></a><span class="lineno"> 2839</span>  max_depth INT,</div> |
| <div class="line"><a name="l02840"></a><span class="lineno"> 2840</span>  tree_id INT</div> |
| <div class="line"><a name="l02841"></a><span class="lineno"> 2841</span>  ) </div> |
| <div class="line"><a name="l02842"></a><span class="lineno"> 2842</span> RETURNS TEXT AS $$ </div> |
| <div class="line"><a name="l02843"></a><span class="lineno"> 2843</span> DECLARE</div> |
| <div class="line"><a name="l02844"></a><span class="lineno"> 2844</span>  ret TEXT := '';</div> |
| <div class="line"><a name="l02845"></a><span class="lineno"> 2845</span>  tree_location INT[];</div> |
| <div class="line"><a name="l02846"></a><span class="lineno"> 2846</span>  feature INT;</div> |
| <div class="line"><a name="l02847"></a><span class="lineno"> 2847</span>  max_class INT;</div> |
| <div class="line"><a name="l02848"></a><span class="lineno"> 2848</span>  num_of_samples INT;</div> |
| <div class="line"><a name="l02849"></a><span class="lineno"> 2849</span>  is_cont BOOLEAN;</div> |
| <div class="line"><a name="l02850"></a><span class="lineno"> 2850</span>  temp_split_value FLOAT;</div> |
| <div class="line"><a name="l02851"></a><span class="lineno"> 2851</span>  index INT;</div> |
| <div class="line"><a name="l02852"></a><span class="lineno"> 2852</span>  curr_value INT;</div> |
| <div class="line"><a name="l02853"></a><span class="lineno"> 2853</span>  probability FLOAT;</div> |
| <div class="line"><a name="l02854"></a><span class="lineno"> 2854</span>  curstmt TEXT;</div> |
| <div class="line"><a name="l02855"></a><span class="lineno"> 2855</span>  child_nid INT;</div> |
| <div class="line"><a name="l02856"></a><span class="lineno"> 2856</span> BEGIN</div> |
| <div class="line"><a name="l02857"></a><span class="lineno"> 2857</span>  IF (id IS NULL OR id <= 0) THEN</div> |
| <div class="line"><a name="l02858"></a><span class="lineno"> 2858</span>  RETURN ret;</div> |
| <div class="line"><a name="l02859"></a><span class="lineno"> 2859</span>  END IF;</div> |
| <div class="line"><a name="l02860"></a><span class="lineno"> 2860</span> </div> |
| <div class="line"><a name="l02861"></a><span class="lineno"> 2861</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02862"></a><span class="lineno"> 2862</span>  (</div> |
| <div class="line"><a name="l02863"></a><span class="lineno"> 2863</span>  'SELECT tree_location, feature, is_cont, </div> |
| <div class="line"><a name="l02864"></a><span class="lineno"> 2864</span>  split_value, max_class,num_of_samples,probability</div> |
| <div class="line"><a name="l02865"></a><span class="lineno"> 2865</span>  FROM %</div> |
| <div class="line"><a name="l02866"></a><span class="lineno"> 2866</span>  WHERE id = % AND tid=%',</div> |
| <div class="line"><a name="l02867"></a><span class="lineno"> 2867</span>  ARRAY[</div> |
| <div class="line"><a name="l02868"></a><span class="lineno"> 2868</span>  tree_table,</div> |
| <div class="line"><a name="l02869"></a><span class="lineno"> 2869</span>  MADLIB_SCHEMA.__to_char(id),</div> |
| <div class="line"><a name="l02870"></a><span class="lineno"> 2870</span>  MADLIB_SCHEMA.__to_char(tree_id)</div> |
| <div class="line"><a name="l02871"></a><span class="lineno"> 2871</span>  ]</div> |
| <div class="line"><a name="l02872"></a><span class="lineno"> 2872</span>  )</div> |
| <div class="line"><a name="l02873"></a><span class="lineno"> 2873</span>  INTO curstmt;</div> |
| <div class="line"><a name="l02874"></a><span class="lineno"> 2874</span>  </div> |
| <div class="line"><a name="l02875"></a><span class="lineno"> 2875</span>  EXECUTE curstmt INTO tree_location, feature, is_cont, </div> |
| <div class="line"><a name="l02876"></a><span class="lineno"> 2876</span>  temp_split_value, max_class, num_of_samples, probability; </div> |
| <div class="line"><a name="l02877"></a><span class="lineno"> 2877</span> </div> |
| <div class="line"><a name="l02878"></a><span class="lineno"> 2878</span>  curr_value = tree_location[array_upper(tree_location,1)];</div> |
| <div class="line"><a name="l02879"></a><span class="lineno"> 2879</span> </div> |
| <div class="line"><a name="l02880"></a><span class="lineno"> 2880</span>  FOR index IN 0..depth LOOP</div> |
| <div class="line"><a name="l02881"></a><span class="lineno"> 2881</span>  ret = ret || ' ';</div> |
| <div class="line"><a name="l02882"></a><span class="lineno"> 2882</span>  END LOOP;</div> |
| <div class="line"><a name="l02883"></a><span class="lineno"> 2883</span>  </div> |
| <div class="line"><a name="l02884"></a><span class="lineno"> 2884</span>  IF (id >tree_id) THEN</div> |
| <div class="line"><a name="l02885"></a><span class="lineno"> 2885</span>  ret = ret ||MADLIB_SCHEMA.__get_feature_name(feature_id,metatable_name)||': ';</div> |
| <div class="line"><a name="l02886"></a><span class="lineno"> 2886</span> </div> |
| <div class="line"><a name="l02887"></a><span class="lineno"> 2887</span>  IF (is_cont) THEN</div> |
| <div class="line"><a name="l02888"></a><span class="lineno"> 2888</span>  IF (curr_value = 1) THEN</div> |
| <div class="line"><a name="l02889"></a><span class="lineno"> 2889</span>  ret = ret || ' <= ';</div> |
| <div class="line"><a name="l02890"></a><span class="lineno"> 2890</span>  ELSE</div> |
| <div class="line"><a name="l02891"></a><span class="lineno"> 2891</span>  ret = ret || ' > ';</div> |
| <div class="line"><a name="l02892"></a><span class="lineno"> 2892</span>  END IF;</div> |
| <div class="line"><a name="l02893"></a><span class="lineno"> 2893</span>  ret = ret || split_value;</div> |
| <div class="line"><a name="l02894"></a><span class="lineno"> 2894</span>  ELSE</div> |
| <div class="line"><a name="l02895"></a><span class="lineno"> 2895</span>  ret = ret || </div> |
| <div class="line"><a name="l02896"></a><span class="lineno"> 2896</span>  ' = ' || </div> |
| <div class="line"><a name="l02897"></a><span class="lineno"> 2897</span>  MADLIB_SCHEMA.__get_feature_value</div> |
| <div class="line"><a name="l02898"></a><span class="lineno"> 2898</span>  (</div> |
| <div class="line"><a name="l02899"></a><span class="lineno"> 2899</span>  feature_id, </div> |
| <div class="line"><a name="l02900"></a><span class="lineno"> 2900</span>  curr_value, </div> |
| <div class="line"><a name="l02901"></a><span class="lineno"> 2901</span>  metatable_name</div> |
| <div class="line"><a name="l02902"></a><span class="lineno"> 2902</span>  );</div> |
| <div class="line"><a name="l02903"></a><span class="lineno"> 2903</span>  END IF;</div> |
| <div class="line"><a name="l02904"></a><span class="lineno"> 2904</span>  ELSE</div> |
| <div class="line"><a name="l02905"></a><span class="lineno"> 2905</span>  ret = ret||'Root Node ';</div> |
| <div class="line"><a name="l02906"></a><span class="lineno"> 2906</span>  END IF;</div> |
| <div class="line"><a name="l02907"></a><span class="lineno"> 2907</span> </div> |
| <div class="line"><a name="l02908"></a><span class="lineno"> 2908</span>  ret = ret || </div> |
| <div class="line"><a name="l02909"></a><span class="lineno"> 2909</span>  ' : class(' || </div> |
| <div class="line"><a name="l02910"></a><span class="lineno"> 2910</span>  MADLIB_SCHEMA.__get_class_value(max_class,metatable_name) || </div> |
| <div class="line"><a name="l02911"></a><span class="lineno"> 2911</span>  ') num_elements(' || </div> |
| <div class="line"><a name="l02912"></a><span class="lineno"> 2912</span>  num_of_samples || </div> |
| <div class="line"><a name="l02913"></a><span class="lineno"> 2913</span>  ') predict_prob(' ||</div> |
| <div class="line"><a name="l02914"></a><span class="lineno"> 2914</span>  probability ||</div> |
| <div class="line"><a name="l02915"></a><span class="lineno"> 2915</span>  ')';</div> |
| <div class="line"><a name="l02916"></a><span class="lineno"> 2916</span> </div> |
| <div class="line"><a name="l02917"></a><span class="lineno"> 2917</span>  ret = ret || E'\n';</div> |
| <div class="line"><a name="l02918"></a><span class="lineno"> 2918</span>  </div> |
| <div class="line"><a name="l02919"></a><span class="lineno"> 2919</span>  IF (max_depth IS NOT NULL AND </div> |
| <div class="line"><a name="l02920"></a><span class="lineno"> 2920</span>  depth >= max_depth) THEN</div> |
| <div class="line"><a name="l02921"></a><span class="lineno"> 2921</span>  RETURN ret;</div> |
| <div class="line"><a name="l02922"></a><span class="lineno"> 2922</span>  END IF;</div> |
| <div class="line"><a name="l02923"></a><span class="lineno"> 2923</span>  </div> |
| <div class="line"><a name="l02924"></a><span class="lineno"> 2924</span>  curstmt = MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l02925"></a><span class="lineno"> 2925</span>  (</div> |
| <div class="line"><a name="l02926"></a><span class="lineno"> 2926</span>  'SELECT id</div> |
| <div class="line"><a name="l02927"></a><span class="lineno"> 2927</span>  FROM %</div> |
| <div class="line"><a name="l02928"></a><span class="lineno"> 2928</span>  WHERE parent_id = % AND tid=%</div> |
| <div class="line"><a name="l02929"></a><span class="lineno"> 2929</span>  ORDER BY id',</div> |
| <div class="line"><a name="l02930"></a><span class="lineno"> 2930</span>  ARRAY[</div> |
| <div class="line"><a name="l02931"></a><span class="lineno"> 2931</span>  tree_table,</div> |
| <div class="line"><a name="l02932"></a><span class="lineno"> 2932</span>  MADLIB_SCHEMA.__to_char(id),</div> |
| <div class="line"><a name="l02933"></a><span class="lineno"> 2933</span>  MADLIB_SCHEMA.__to_char(tree_id)</div> |
| <div class="line"><a name="l02934"></a><span class="lineno"> 2934</span>  ]</div> |
| <div class="line"><a name="l02935"></a><span class="lineno"> 2935</span>  );</div> |
| <div class="line"><a name="l02936"></a><span class="lineno"> 2936</span> </div> |
| <div class="line"><a name="l02937"></a><span class="lineno"> 2937</span>  FOR child_nid IN EXECUTE curstmt LOOP</div> |
| <div class="line"><a name="l02938"></a><span class="lineno"> 2938</span>  ret = ret || MADLIB_SCHEMA.__display_tree_no_ordered_aggr(</div> |
| <div class="line"><a name="l02939"></a><span class="lineno"> 2939</span>  tree_table,</div> |
| <div class="line"><a name="l02940"></a><span class="lineno"> 2940</span>  child_nid,</div> |
| <div class="line"><a name="l02941"></a><span class="lineno"> 2941</span>  feature,</div> |
| <div class="line"><a name="l02942"></a><span class="lineno"> 2942</span>  depth + 1,</div> |
| <div class="line"><a name="l02943"></a><span class="lineno"> 2943</span>  is_cont,</div> |
| <div class="line"><a name="l02944"></a><span class="lineno"> 2944</span>  temp_split_value,</div> |
| <div class="line"><a name="l02945"></a><span class="lineno"> 2945</span>  metatable_name,</div> |
| <div class="line"><a name="l02946"></a><span class="lineno"> 2946</span>  max_depth,</div> |
| <div class="line"><a name="l02947"></a><span class="lineno"> 2947</span>  tree_id); </div> |
| <div class="line"><a name="l02948"></a><span class="lineno"> 2948</span>  END LOOP;</div> |
| <div class="line"><a name="l02949"></a><span class="lineno"> 2949</span> </div> |
| <div class="line"><a name="l02950"></a><span class="lineno"> 2950</span>  RETURN ret;</div> |
| <div class="line"><a name="l02951"></a><span class="lineno"> 2951</span> END $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l02952"></a><span class="lineno"> 2952</span> </div> |
| <div class="line"><a name="l02953"></a><span class="lineno"> 2953</span> </div> |
| <div class="line"><a name="l02954"></a><span class="lineno"> 2954</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l02955"></a><span class="lineno"> 2955</span> <span class="comment"> * @brief Display the trained model with human readable format. It use the </span></div> |
| <div class="line"><a name="l02956"></a><span class="lineno"> 2956</span> <span class="comment"> * recursive algorithm, which is slower than the version with </span></div> |
| <div class="line"><a name="l02957"></a><span class="lineno"> 2957</span> <span class="comment"> * ordered aggregate. We only use it when ordered aggregate is unavailable.</span></div> |
| <div class="line"><a name="l02958"></a><span class="lineno"> 2958</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l02959"></a><span class="lineno"> 2959</span> <span class="comment"> * @param tree_table The full name of the tree table. </span></div> |
| <div class="line"><a name="l02960"></a><span class="lineno"> 2960</span> <span class="comment"> * @param tree_id The array contains the IDs of the trees to be displayed.</span></div> |
| <div class="line"><a name="l02961"></a><span class="lineno"> 2961</span> <span class="comment"> * @param max_depth The max depth to be displayed. If it is set to null,</span></div> |
| <div class="line"><a name="l02962"></a><span class="lineno"> 2962</span> <span class="comment"> * this function will show all levels. </span></div> |
| <div class="line"><a name="l02963"></a><span class="lineno"> 2963</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l02964"></a><span class="lineno"> 2964</span> <span class="comment"> * @return The text representing the tree with human readable format.</span></div> |
| <div class="line"><a name="l02965"></a><span class="lineno"> 2965</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l02966"></a><span class="lineno"> 2966</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l02967"></a><span class="lineno"> 2967</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__treemodel_display_no_ordered_aggr</div> |
| <div class="line"><a name="l02968"></a><span class="lineno"> 2968</span>  (</div> |
| <div class="line"><a name="l02969"></a><span class="lineno"> 2969</span>  tree_table TEXT,</div> |
| <div class="line"><a name="l02970"></a><span class="lineno"> 2970</span>  tree_id INT[],</div> |
| <div class="line"><a name="l02971"></a><span class="lineno"> 2971</span>  max_depth INT</div> |
| <div class="line"><a name="l02972"></a><span class="lineno"> 2972</span>  ) </div> |
| <div class="line"><a name="l02973"></a><span class="lineno"> 2973</span> RETURNS SETOF TEXT AS $$</div> |
| <div class="line"><a name="l02974"></a><span class="lineno"> 2974</span> DECLARE</div> |
| <div class="line"><a name="l02975"></a><span class="lineno"> 2975</span>  metatable_name TEXT := null;</div> |
| <div class="line"><a name="l02976"></a><span class="lineno"> 2976</span>  curstmt TEXT := '';</div> |
| <div class="line"><a name="l02977"></a><span class="lineno"> 2977</span>  index INT;</div> |
| <div class="line"><a name="l02978"></a><span class="lineno"> 2978</span>  result TEXT := '';</div> |
| <div class="line"><a name="l02979"></a><span class="lineno"> 2979</span>  root_id INT;</div> |
| <div class="line"><a name="l02980"></a><span class="lineno"> 2980</span> BEGIN</div> |
| <div class="line"><a name="l02981"></a><span class="lineno"> 2981</span>  PERFORM MADLIB_SCHEMA.__assert_table</div> |
| <div class="line"><a name="l02982"></a><span class="lineno"> 2982</span>  (</div> |
| <div class="line"><a name="l02983"></a><span class="lineno"> 2983</span>  tree_table,</div> |
| <div class="line"><a name="l02984"></a><span class="lineno"> 2984</span>  't'</div> |
| <div class="line"><a name="l02985"></a><span class="lineno"> 2985</span>  ); </div> |
| <div class="line"><a name="l02986"></a><span class="lineno"> 2986</span>  </div> |
| <div class="line"><a name="l02987"></a><span class="lineno"> 2987</span>  metatable_name = MADLIB_SCHEMA.__get_metatable_name( tree_table );</div> |
| <div class="line"><a name="l02988"></a><span class="lineno"> 2988</span> </div> |
| <div class="line"><a name="l02989"></a><span class="lineno"> 2989</span>  index= array_lower(tree_id,1);</div> |
| <div class="line"><a name="l02990"></a><span class="lineno"> 2990</span>  </div> |
| <div class="line"><a name="l02991"></a><span class="lineno"> 2991</span>  WHILE (index<=array_upper(tree_id,1) ) LOOP</div> |
| <div class="line"><a name="l02992"></a><span class="lineno"> 2992</span>  EXECUTE 'SELECT id FROM '||tree_table||</div> |
| <div class="line"><a name="l02993"></a><span class="lineno"> 2993</span>  ' WHERE parent_id=0 and tid='||tree_id[index]||';' INTO root_id; </div> |
| <div class="line"><a name="l02994"></a><span class="lineno"> 2994</span> </div> |
| <div class="line"><a name="l02995"></a><span class="lineno"> 2995</span>  RETURN NEXT E'\nTree '||tree_id[index]||E'\n'||</div> |
| <div class="line"><a name="l02996"></a><span class="lineno"> 2996</span>  MADLIB_SCHEMA.__display_tree_no_ordered_aggr(tree_table, root_id, 0, 0, 'f', </div> |
| <div class="line"><a name="l02997"></a><span class="lineno"> 2997</span>  0, metatable_name,max_depth,tree_id[index]);</div> |
| <div class="line"><a name="l02998"></a><span class="lineno"> 2998</span>  index=index+1;</div> |
| <div class="line"><a name="l02999"></a><span class="lineno"> 2999</span>  END LOOP;</div> |
| <div class="line"><a name="l03000"></a><span class="lineno"> 3000</span>  RETURN;</div> |
| <div class="line"><a name="l03001"></a><span class="lineno"> 3001</span> END $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l03002"></a><span class="lineno"> 3002</span> </div> |
| <div class="line"><a name="l03003"></a><span class="lineno"> 3003</span> </div> |
| <div class="line"><a name="l03004"></a><span class="lineno"> 3004</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l03005"></a><span class="lineno"> 3005</span> <span class="comment"> * @brief Multiple trees may classify the same record to different classes. </span></div> |
| <div class="line"><a name="l03006"></a><span class="lineno"> 3006</span> <span class="comment"> * This function gets the results voted by multiple trees.</span></div> |
| <div class="line"><a name="l03007"></a><span class="lineno"> 3007</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03008"></a><span class="lineno"> 3008</span> <span class="comment"> * @param src_table The full name of the table containing original data.</span></div> |
| <div class="line"><a name="l03009"></a><span class="lineno"> 3009</span> <span class="comment"> * @param dst_table The full name of the table to store the voted results. </span></div> |
| <div class="line"><a name="l03010"></a><span class="lineno"> 3010</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03011"></a><span class="lineno"> 3011</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l03012"></a><span class="lineno"> 3012</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__treemodel_get_vote_result</div> |
| <div class="line"><a name="l03013"></a><span class="lineno"> 3013</span>  (</div> |
| <div class="line"><a name="l03014"></a><span class="lineno"> 3014</span>  src_table TEXT, </div> |
| <div class="line"><a name="l03015"></a><span class="lineno"> 3015</span>  dst_table TEXT</div> |
| <div class="line"><a name="l03016"></a><span class="lineno"> 3016</span>  )</div> |
| <div class="line"><a name="l03017"></a><span class="lineno"> 3017</span> RETURNS VOID AS $$</div> |
| <div class="line"><a name="l03018"></a><span class="lineno"> 3018</span> DECLARE</div> |
| <div class="line"><a name="l03019"></a><span class="lineno"> 3019</span>  curstmt TEXT;</div> |
| <div class="line"><a name="l03020"></a><span class="lineno"> 3020</span> BEGIN</div> |
| <div class="line"><a name="l03021"></a><span class="lineno"> 3021</span>  EXECUTE 'DROP TABLE IF EXISTS '||dst_table;</div> |
| <div class="line"><a name="l03022"></a><span class="lineno"> 3022</span>  EXECUTE 'CREATE TEMP TABLE '||dst_table||E'</div> |
| <div class="line"><a name="l03023"></a><span class="lineno"> 3023</span>  (</div> |
| <div class="line"><a name="l03024"></a><span class="lineno"> 3024</span>  id BIGINT,</div> |
| <div class="line"><a name="l03025"></a><span class="lineno"> 3025</span>  class INT,</div> |
| <div class="line"><a name="l03026"></a><span class="lineno"> 3026</span>  prob FLOAT8</div> |
| <div class="line"><a name="l03027"></a><span class="lineno"> 3027</span>  )m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (id)');';</div> |
| <div class="line"><a name="l03028"></a><span class="lineno"> 3028</span> </div> |
| <div class="line"><a name="l03029"></a><span class="lineno"> 3029</span>  SELECT MADLIB_SCHEMA.__format(</div> |
| <div class="line"><a name="l03030"></a><span class="lineno"> 3030</span>  'INSERT INTO %</div> |
| <div class="line"><a name="l03031"></a><span class="lineno"> 3031</span>  SELECT id, max_array[3], max_array[2] FROM </div> |
| <div class="line"><a name="l03032"></a><span class="lineno"> 3032</span>  (SELECT id, max(array[count,prob,class]) AS max_array FROM</div> |
| <div class="line"><a name="l03033"></a><span class="lineno"> 3033</span>  (SELECT id, class, COUNT(*) AS count, AVG(prob) as prob FROM</div> |
| <div class="line"><a name="l03034"></a><span class="lineno"> 3034</span>  % GROUP BY id,class) t1 GROUP BY id) t2',</div> |
| <div class="line"><a name="l03035"></a><span class="lineno"> 3035</span>  ARRAY[</div> |
| <div class="line"><a name="l03036"></a><span class="lineno"> 3036</span>  dst_table,</div> |
| <div class="line"><a name="l03037"></a><span class="lineno"> 3037</span>  src_table</div> |
| <div class="line"><a name="l03038"></a><span class="lineno"> 3038</span>  ]</div> |
| <div class="line"><a name="l03039"></a><span class="lineno"> 3039</span>  ) </div> |
| <div class="line"><a name="l03040"></a><span class="lineno"> 3040</span>  INTO curstmt;</div> |
| <div class="line"><a name="l03041"></a><span class="lineno"> 3041</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l03042"></a><span class="lineno"> 3042</span>  RETURN;</div> |
| <div class="line"><a name="l03043"></a><span class="lineno"> 3043</span> END</div> |
| <div class="line"><a name="l03044"></a><span class="lineno"> 3044</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l03045"></a><span class="lineno"> 3045</span> </div> |
| <div class="line"><a name="l03046"></a><span class="lineno"> 3046</span> </div> |
| <div class="line"><a name="l03047"></a><span class="lineno"> 3047</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l03048"></a><span class="lineno"> 3048</span> <span class="comment"> * @brief An internal classification function. It classifies with all trees at </span></div> |
| <div class="line"><a name="l03049"></a><span class="lineno"> 3049</span> <span class="comment"> * the same time. For medium/small data sets, tests shows that it is more</span></div> |
| <div class="line"><a name="l03050"></a><span class="lineno"> 3050</span> <span class="comment"> * efficient than the serial classification function. </span></div> |
| <div class="line"><a name="l03051"></a><span class="lineno"> 3051</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03052"></a><span class="lineno"> 3052</span> <span class="comment"> * @param classification_table_name The full name of the table containing the </span></div> |
| <div class="line"><a name="l03053"></a><span class="lineno"> 3053</span> <span class="comment"> * classification set.</span></div> |
| <div class="line"><a name="l03054"></a><span class="lineno"> 3054</span> <span class="comment"> * @param tree_table_name The full name of the tree table.</span></div> |
| <div class="line"><a name="l03055"></a><span class="lineno"> 3055</span> <span class="comment"> * @param verbosity > 0 means this function runs in verbose mode. </span></div> |
| <div class="line"><a name="l03056"></a><span class="lineno"> 3056</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03057"></a><span class="lineno"> 3057</span> <span class="comment"> * @return An array containing the encoded table name and classification result </span></div> |
| <div class="line"><a name="l03058"></a><span class="lineno"> 3058</span> <span class="comment"> * table name (We encode the source table during the classification).</span></div> |
| <div class="line"><a name="l03059"></a><span class="lineno"> 3059</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03060"></a><span class="lineno"> 3060</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l03061"></a><span class="lineno"> 3061</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__treemodel_classify_internal</div> |
| <div class="line"><a name="l03062"></a><span class="lineno"> 3062</span>  (</div> |
| <div class="line"><a name="l03063"></a><span class="lineno"> 3063</span>  classification_table_name TEXT, </div> |
| <div class="line"><a name="l03064"></a><span class="lineno"> 3064</span>  tree_table_name TEXT, </div> |
| <div class="line"><a name="l03065"></a><span class="lineno"> 3065</span>  verbosity INT</div> |
| <div class="line"><a name="l03066"></a><span class="lineno"> 3066</span>  ) </div> |
| <div class="line"><a name="l03067"></a><span class="lineno"> 3067</span> RETURNS TEXT[] AS $$</div> |
| <div class="line"><a name="l03068"></a><span class="lineno"> 3068</span> DECLARE</div> |
| <div class="line"><a name="l03069"></a><span class="lineno"> 3069</span>  table_pick INT := 1;</div> |
| <div class="line"><a name="l03070"></a><span class="lineno"> 3070</span>  remains_to_classify INT;</div> |
| <div class="line"><a name="l03071"></a><span class="lineno"> 3071</span>  size_finished INT;</div> |
| <div class="line"><a name="l03072"></a><span class="lineno"> 3072</span>  time_stamp TIMESTAMP;</div> |
| <div class="line"><a name="l03073"></a><span class="lineno"> 3073</span>  metatable_name TEXT := '';</div> |
| <div class="line"><a name="l03074"></a><span class="lineno"> 3074</span>  id_col_name TEXT := 'id';</div> |
| <div class="line"><a name="l03075"></a><span class="lineno"> 3075</span>  curr_level INT := 1;</div> |
| <div class="line"><a name="l03076"></a><span class="lineno"> 3076</span>  max_level INT := 0;</div> |
| <div class="line"><a name="l03077"></a><span class="lineno"> 3077</span>  h2hmv_routine_id INT := 0;</div> |
| <div class="line"><a name="l03078"></a><span class="lineno"> 3078</span>  curstmt TEXT := '';</div> |
| <div class="line"><a name="l03079"></a><span class="lineno"> 3079</span>  result_table_name TEXT := 'dt_classify_internal_rt';</div> |
| <div class="line"><a name="l03080"></a><span class="lineno"> 3080</span>  encoded_table_name TEXT := 'dt_classify_internal_edt';</div> |
| <div class="line"><a name="l03081"></a><span class="lineno"> 3081</span>  table_names TEXT[] := '{classified_instance_ping,classified_instance_pong}';</div> |
| <div class="line"><a name="l03082"></a><span class="lineno"> 3082</span>  tree_id INT;</div> |
| <div class="line"><a name="l03083"></a><span class="lineno"> 3083</span> BEGIN</div> |
| <div class="line"><a name="l03084"></a><span class="lineno"> 3084</span>  time_stamp = clock_timestamp();</div> |
| <div class="line"><a name="l03085"></a><span class="lineno"> 3085</span> </div> |
| <div class="line"><a name="l03086"></a><span class="lineno"> 3086</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03087"></a><span class="lineno"> 3087</span>  (</div> |
| <div class="line"><a name="l03088"></a><span class="lineno"> 3088</span>  (classification_table_name IS NOT NULL) AND</div> |
| <div class="line"><a name="l03089"></a><span class="lineno"> 3089</span>  (</div> |
| <div class="line"><a name="l03090"></a><span class="lineno"> 3090</span>  MADLIB_SCHEMA.__table_exists</div> |
| <div class="line"><a name="l03091"></a><span class="lineno"> 3091</span>  (</div> |
| <div class="line"><a name="l03092"></a><span class="lineno"> 3092</span>  classification_table_name</div> |
| <div class="line"><a name="l03093"></a><span class="lineno"> 3093</span>  )</div> |
| <div class="line"><a name="l03094"></a><span class="lineno"> 3094</span>  ),</div> |
| <div class="line"><a name="l03095"></a><span class="lineno"> 3095</span>  'the specified classification table' || </div> |
| <div class="line"><a name="l03096"></a><span class="lineno"> 3096</span>  coalesce('<' || classification_table_name || </div> |
| <div class="line"><a name="l03097"></a><span class="lineno"> 3097</span>  '> does not exists', ' is NULL')</div> |
| <div class="line"><a name="l03098"></a><span class="lineno"> 3098</span>  ); </div> |
| <div class="line"><a name="l03099"></a><span class="lineno"> 3099</span> </div> |
| <div class="line"><a name="l03100"></a><span class="lineno"> 3100</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03101"></a><span class="lineno"> 3101</span>  (</div> |
| <div class="line"><a name="l03102"></a><span class="lineno"> 3102</span>  (tree_table_name IS NOT NULL) AND</div> |
| <div class="line"><a name="l03103"></a><span class="lineno"> 3103</span>  (</div> |
| <div class="line"><a name="l03104"></a><span class="lineno"> 3104</span>  MADLIB_SCHEMA.__table_exists</div> |
| <div class="line"><a name="l03105"></a><span class="lineno"> 3105</span>  (</div> |
| <div class="line"><a name="l03106"></a><span class="lineno"> 3106</span>  tree_table_name</div> |
| <div class="line"><a name="l03107"></a><span class="lineno"> 3107</span>  )</div> |
| <div class="line"><a name="l03108"></a><span class="lineno"> 3108</span>  ),</div> |
| <div class="line"><a name="l03109"></a><span class="lineno"> 3109</span>  'the specified tree table' || </div> |
| <div class="line"><a name="l03110"></a><span class="lineno"> 3110</span>  coalesce('<' || tree_table_name || '> does not exists', ' is NULL')</div> |
| <div class="line"><a name="l03111"></a><span class="lineno"> 3111</span>  ); </div> |
| <div class="line"><a name="l03112"></a><span class="lineno"> 3112</span>  </div> |
| <div class="line"><a name="l03113"></a><span class="lineno"> 3113</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03114"></a><span class="lineno"> 3114</span>  (</div> |
| <div class="line"><a name="l03115"></a><span class="lineno"> 3115</span>  verbosity IS NOT NULL, </div> |
| <div class="line"><a name="l03116"></a><span class="lineno"> 3116</span>  'verbosity must be non-null'</div> |
| <div class="line"><a name="l03117"></a><span class="lineno"> 3117</span>  ); </div> |
| <div class="line"><a name="l03118"></a><span class="lineno"> 3118</span> </div> |
| <div class="line"><a name="l03119"></a><span class="lineno"> 3119</span>  EXECUTE 'DROP TABLE IF EXISTS ' || encoded_table_name || ' CASCADE';</div> |
| <div class="line"><a name="l03120"></a><span class="lineno"> 3120</span>  </div> |
| <div class="line"><a name="l03121"></a><span class="lineno"> 3121</span>  SELECT MADLIB_SCHEMA.__get_metatable_name(tree_table_name) INTO metatable_name;</div> |
| <div class="line"><a name="l03122"></a><span class="lineno"> 3122</span> </div> |
| <div class="line"><a name="l03123"></a><span class="lineno"> 3123</span>  SELECT MADLIB_SCHEMA.__get_routine_id(tree_table_name) INTO h2hmv_routine_id;</div> |
| <div class="line"><a name="l03124"></a><span class="lineno"> 3124</span>  </div> |
| <div class="line"><a name="l03125"></a><span class="lineno"> 3125</span>  PERFORM MADLIB_SCHEMA.__encode_table</div> |
| <div class="line"><a name="l03126"></a><span class="lineno"> 3126</span>  (</div> |
| <div class="line"><a name="l03127"></a><span class="lineno"> 3127</span>  classification_table_name, </div> |
| <div class="line"><a name="l03128"></a><span class="lineno"> 3128</span>  encoded_table_name, </div> |
| <div class="line"><a name="l03129"></a><span class="lineno"> 3129</span>  metatable_name, </div> |
| <div class="line"><a name="l03130"></a><span class="lineno"> 3130</span>  h2hmv_routine_id,</div> |
| <div class="line"><a name="l03131"></a><span class="lineno"> 3131</span>  verbosity</div> |
| <div class="line"><a name="l03132"></a><span class="lineno"> 3132</span>  );</div> |
| <div class="line"><a name="l03133"></a><span class="lineno"> 3133</span>  </div> |
| <div class="line"><a name="l03134"></a><span class="lineno"> 3134</span>  IF (verbosity > 0) THEN</div> |
| <div class="line"><a name="l03135"></a><span class="lineno"> 3135</span>  RAISE INFO 'tabular format. id_col_name: %', id_col_name;</div> |
| <div class="line"><a name="l03136"></a><span class="lineno"> 3136</span>  END IF; </div> |
| <div class="line"><a name="l03137"></a><span class="lineno"> 3137</span>  </div> |
| <div class="line"><a name="l03138"></a><span class="lineno"> 3138</span>  <span class="comment">/*</span></div> |
| <div class="line"><a name="l03139"></a><span class="lineno"> 3139</span> <span class="comment"> * The table of classified_instance_ping and classified_instance_pong are</span></div> |
| <div class="line"><a name="l03140"></a><span class="lineno"> 3140</span> <span class="comment"> * auxiliary tables used during the classification process.</span></div> |
| <div class="line"><a name="l03141"></a><span class="lineno"> 3141</span> <span class="comment"> * For each record, these tables tell us which node it belongs to. They also</span></div> |
| <div class="line"><a name="l03142"></a><span class="lineno"> 3142</span> <span class="comment"> * hold the information of class and probability.</span></div> |
| <div class="line"><a name="l03143"></a><span class="lineno"> 3143</span> <span class="comment"> * We use transfer data between these two tables rather than update a single</span></div> |
| <div class="line"><a name="l03144"></a><span class="lineno"> 3144</span> <span class="comment"> * table during the classification process. We find the operation of update</span></div> |
| <div class="line"><a name="l03145"></a><span class="lineno"> 3145</span> <span class="comment"> * is quite expensive.</span></div> |
| <div class="line"><a name="l03146"></a><span class="lineno"> 3146</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l03147"></a><span class="lineno"> 3147</span>  DROP TABLE IF EXISTS classified_instance_ping;</div> |
| <div class="line"><a name="l03148"></a><span class="lineno"> 3148</span>  CREATE TEMP TABLE classified_instance_ping</div> |
| <div class="line"><a name="l03149"></a><span class="lineno"> 3149</span>  (</div> |
| <div class="line"><a name="l03150"></a><span class="lineno"> 3150</span>  tid INT,</div> |
| <div class="line"><a name="l03151"></a><span class="lineno"> 3151</span>  id BIGINT,</div> |
| <div class="line"><a name="l03152"></a><span class="lineno"> 3152</span>  jump INT,</div> |
| <div class="line"><a name="l03153"></a><span class="lineno"> 3153</span>  class INT,</div> |
| <div class="line"><a name="l03154"></a><span class="lineno"> 3154</span>  prob FLOAT,</div> |
| <div class="line"><a name="l03155"></a><span class="lineno"> 3155</span>  parent_id INT,</div> |
| <div class="line"><a name="l03156"></a><span class="lineno"> 3156</span>  leaf_id INT</div> |
| <div class="line"><a name="l03157"></a><span class="lineno"> 3157</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (id)');</div> |
| <div class="line"><a name="l03158"></a><span class="lineno"> 3158</span>  </div> |
| <div class="line"><a name="l03159"></a><span class="lineno"> 3159</span>  DROP TABLE IF EXISTS classified_instance_pong;</div> |
| <div class="line"><a name="l03160"></a><span class="lineno"> 3160</span>  CREATE TEMP TABLE classified_instance_pong</div> |
| <div class="line"><a name="l03161"></a><span class="lineno"> 3161</span>  (</div> |
| <div class="line"><a name="l03162"></a><span class="lineno"> 3162</span>  tid INT,</div> |
| <div class="line"><a name="l03163"></a><span class="lineno"> 3163</span>  id BIGINT,</div> |
| <div class="line"><a name="l03164"></a><span class="lineno"> 3164</span>  jump INT,</div> |
| <div class="line"><a name="l03165"></a><span class="lineno"> 3165</span>  class INT,</div> |
| <div class="line"><a name="l03166"></a><span class="lineno"> 3166</span>  prob FLOAT,</div> |
| <div class="line"><a name="l03167"></a><span class="lineno"> 3167</span>  parent_id INT,</div> |
| <div class="line"><a name="l03168"></a><span class="lineno"> 3168</span>  leaf_id INT</div> |
| <div class="line"><a name="l03169"></a><span class="lineno"> 3169</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (id)');</div> |
| <div class="line"><a name="l03170"></a><span class="lineno"> 3170</span> </div> |
| <div class="line"><a name="l03171"></a><span class="lineno"> 3171</span> </div> |
| <div class="line"><a name="l03172"></a><span class="lineno"> 3172</span>  EXECUTE 'DROP TABLE IF EXISTS ' || result_table_name || ' CASCADE';</div> |
| <div class="line"><a name="l03173"></a><span class="lineno"> 3173</span>  EXECUTE 'CREATE TEMP TABLE ' || result_table_name || E'</div> |
| <div class="line"><a name="l03174"></a><span class="lineno"> 3174</span>  (</div> |
| <div class="line"><a name="l03175"></a><span class="lineno"> 3175</span>  tid INT,</div> |
| <div class="line"><a name="l03176"></a><span class="lineno"> 3176</span>  id BIGINT,</div> |
| <div class="line"><a name="l03177"></a><span class="lineno"> 3177</span>  jump INT,</div> |
| <div class="line"><a name="l03178"></a><span class="lineno"> 3178</span>  class INT,</div> |
| <div class="line"><a name="l03179"></a><span class="lineno"> 3179</span>  prob FLOAT,</div> |
| <div class="line"><a name="l03180"></a><span class="lineno"> 3180</span>  parent_id INT,</div> |
| <div class="line"><a name="l03181"></a><span class="lineno"> 3181</span>  leaf_id INT</div> |
| <div class="line"><a name="l03182"></a><span class="lineno"> 3182</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (id)');';</div> |
| <div class="line"><a name="l03183"></a><span class="lineno"> 3183</span> </div> |
| <div class="line"><a name="l03184"></a><span class="lineno"> 3184</span> </div> |
| <div class="line"><a name="l03185"></a><span class="lineno"> 3185</span>  EXECUTE 'INSERT INTO classified_instance_ping (id, jump, class, prob,tid) </div> |
| <div class="line"><a name="l03186"></a><span class="lineno"> 3186</span>  SELECT m.'||id_col_name||', t.id, 0, 0, t.tid </div> |
| <div class="line"><a name="l03187"></a><span class="lineno"> 3187</span>  FROM ' || encoded_table_name || ' m CROSS JOIN </div> |
| <div class="line"><a name="l03188"></a><span class="lineno"> 3188</span>  (SELECT DISTINCT tid,id FROM '||tree_table_name||' WHERE parent_id=0) t;'; </div> |
| <div class="line"><a name="l03189"></a><span class="lineno"> 3189</span> </div> |
| <div class="line"><a name="l03190"></a><span class="lineno"> 3190</span>  </div> |
| <div class="line"><a name="l03191"></a><span class="lineno"> 3191</span>  EXECUTE 'SELECT max(array_upper(tree_location,1)) FROM '||tree_table_name||';' </div> |
| <div class="line"><a name="l03192"></a><span class="lineno"> 3192</span>  INTO max_level;</div> |
| <div class="line"><a name="l03193"></a><span class="lineno"> 3193</span> </div> |
| <div class="line"><a name="l03194"></a><span class="lineno"> 3194</span>  IF( max_level is NULL ) THEN</div> |
| <div class="line"><a name="l03195"></a><span class="lineno"> 3195</span>  RAISE EXCEPTION 'tree should not be empty';</div> |
| <div class="line"><a name="l03196"></a><span class="lineno"> 3196</span>  END IF;</div> |
| <div class="line"><a name="l03197"></a><span class="lineno"> 3197</span> </div> |
| <div class="line"><a name="l03198"></a><span class="lineno"> 3198</span>  FOR curr_level IN 1..max_level LOOP</div> |
| <div class="line"><a name="l03199"></a><span class="lineno"> 3199</span>  IF (verbosity > 0) THEN </div> |
| <div class="line"><a name="l03200"></a><span class="lineno"> 3200</span>  RAISE INFO 'new_depth: %', curr_level;</div> |
| <div class="line"><a name="l03201"></a><span class="lineno"> 3201</span>  END IF;</div> |
| <div class="line"><a name="l03202"></a><span class="lineno"> 3202</span>  </div> |
| <div class="line"><a name="l03203"></a><span class="lineno"> 3203</span>  table_pick = table_pick % 2 + 1; </div> |
| <div class="line"><a name="l03204"></a><span class="lineno"> 3204</span>  </div> |
| <div class="line"><a name="l03205"></a><span class="lineno"> 3205</span>  EXECUTE 'TRUNCATE '|| table_names[table_pick] ||';';</div> |
| <div class="line"><a name="l03206"></a><span class="lineno"> 3206</span>  EXECUTE 'SELECT count(id) FROM '||result_table_name||';' INTO size_finished;</div> |
| <div class="line"><a name="l03207"></a><span class="lineno"> 3207</span>  </div> |
| <div class="line"><a name="l03208"></a><span class="lineno"> 3208</span>  IF (verbosity > 0) THEN </div> |
| <div class="line"><a name="l03209"></a><span class="lineno"> 3209</span>  RAISE INFO 'size_finished %', size_finished;</div> |
| <div class="line"><a name="l03210"></a><span class="lineno"> 3210</span>  END IF; </div> |
| <div class="line"><a name="l03211"></a><span class="lineno"> 3211</span>  </div> |
| <div class="line"><a name="l03212"></a><span class="lineno"> 3212</span>  EXECUTE 'SELECT count(*) FROM '|| table_names[(table_pick) % 2 + 1] ||';' </div> |
| <div class="line"><a name="l03213"></a><span class="lineno"> 3213</span>  INTO remains_to_classify;</div> |
| <div class="line"><a name="l03214"></a><span class="lineno"> 3214</span>  </div> |
| <div class="line"><a name="l03215"></a><span class="lineno"> 3215</span>  IF (remains_to_classify = 0) THEN</div> |
| <div class="line"><a name="l03216"></a><span class="lineno"> 3216</span>  IF (verbosity > 0) THEN </div> |
| <div class="line"><a name="l03217"></a><span class="lineno"> 3217</span>  RAISE INFO 'size_finished: % remains_to_classify: %', </div> |
| <div class="line"><a name="l03218"></a><span class="lineno"> 3218</span>  size_finished, remains_to_classify;</div> |
| <div class="line"><a name="l03219"></a><span class="lineno"> 3219</span>  END IF; </div> |
| <div class="line"><a name="l03220"></a><span class="lineno"> 3220</span>  </div> |
| <div class="line"><a name="l03221"></a><span class="lineno"> 3221</span>  EXIT;</div> |
| <div class="line"><a name="l03222"></a><span class="lineno"> 3222</span>  END IF;</div> |
| <div class="line"><a name="l03223"></a><span class="lineno"> 3223</span>  </div> |
| <div class="line"><a name="l03224"></a><span class="lineno"> 3224</span>  SELECT MADLIB_SCHEMA.__format(</div> |
| <div class="line"><a name="l03225"></a><span class="lineno"> 3225</span>  'INSERT INTO %</div> |
| <div class="line"><a name="l03226"></a><span class="lineno"> 3226</span>  SELECT pt.tid, pt.id, </div> |
| <div class="line"><a name="l03227"></a><span class="lineno"> 3227</span>  CASE WHEN (is_cont) THEN </div> |
| <div class="line"><a name="l03228"></a><span class="lineno"> 3228</span>  CASE WHEN (gt.lmc_nid IS NULL) THEN</div> |
| <div class="line"><a name="l03229"></a><span class="lineno"> 3229</span>  0</div> |
| <div class="line"><a name="l03230"></a><span class="lineno"> 3230</span>  ELSE</div> |
| <div class="line"><a name="l03231"></a><span class="lineno"> 3231</span>  gt.lmc_nid + </div> |
| <div class="line"><a name="l03232"></a><span class="lineno"> 3232</span>  float8lt(gt.split_value, fvals[gt.feature])::INT4 + 1 -</div> |
| <div class="line"><a name="l03233"></a><span class="lineno"> 3233</span>  gt.lmc_fval</div> |
| <div class="line"><a name="l03234"></a><span class="lineno"> 3234</span>  END</div> |
| <div class="line"><a name="l03235"></a><span class="lineno"> 3235</span>  ELSE </div> |
| <div class="line"><a name="l03236"></a><span class="lineno"> 3236</span>  CASE WHEN (gt.lmc_nid IS NULL) THEN</div> |
| <div class="line"><a name="l03237"></a><span class="lineno"> 3237</span>  0</div> |
| <div class="line"><a name="l03238"></a><span class="lineno"> 3238</span>  ELSE </div> |
| <div class="line"><a name="l03239"></a><span class="lineno"> 3239</span>  gt.lmc_nid + fvals[gt.feature] - gt.lmc_fval</div> |
| <div class="line"><a name="l03240"></a><span class="lineno"> 3240</span>  END</div> |
| <div class="line"><a name="l03241"></a><span class="lineno"> 3241</span>  END as newjump,</div> |
| <div class="line"><a name="l03242"></a><span class="lineno"> 3242</span>  gt.max_class, gt.probability, gt.parent_id, gt.id </div> |
| <div class="line"><a name="l03243"></a><span class="lineno"> 3243</span>  FROM </div> |
| <div class="line"><a name="l03244"></a><span class="lineno"> 3244</span>  (SELECT t1.tid, t1.id, t1.jump, fvals </div> |
| <div class="line"><a name="l03245"></a><span class="lineno"> 3245</span>  FROM % t1 </div> |
| <div class="line"><a name="l03246"></a><span class="lineno"> 3246</span>  LEFT JOIN % t2 </div> |
| <div class="line"><a name="l03247"></a><span class="lineno"> 3247</span>  ON t1.id = t2.id) AS pt, </div> |
| <div class="line"><a name="l03248"></a><span class="lineno"> 3248</span>  (SELECT tid,lmc_nid, lmc_fval, max_class,feature, probability, </div> |
| <div class="line"><a name="l03249"></a><span class="lineno"> 3249</span>  parent_id, id, is_cont, split_value</div> |
| <div class="line"><a name="l03250"></a><span class="lineno"> 3250</span>  FROM % </div> |
| <div class="line"><a name="l03251"></a><span class="lineno"> 3251</span>  WHERE array_upper(tree_location,1) = %) AS gt </div> |
| <div class="line"><a name="l03252"></a><span class="lineno"> 3252</span>  WHERE pt.jump = gt.id AND pt.tid=gt.tid;',</div> |
| <div class="line"><a name="l03253"></a><span class="lineno"> 3253</span>  ARRAY[</div> |
| <div class="line"><a name="l03254"></a><span class="lineno"> 3254</span>  table_names[table_pick],</div> |
| <div class="line"><a name="l03255"></a><span class="lineno"> 3255</span>  table_names[(table_pick) % 2 + 1],</div> |
| <div class="line"><a name="l03256"></a><span class="lineno"> 3256</span>  encoded_table_name,</div> |
| <div class="line"><a name="l03257"></a><span class="lineno"> 3257</span>  tree_table_name,</div> |
| <div class="line"><a name="l03258"></a><span class="lineno"> 3258</span>  MADLIB_SCHEMA.__to_char(curr_level)</div> |
| <div class="line"><a name="l03259"></a><span class="lineno"> 3259</span>  ]</div> |
| <div class="line"><a name="l03260"></a><span class="lineno"> 3260</span>  )</div> |
| <div class="line"><a name="l03261"></a><span class="lineno"> 3261</span>  INTO curstmt; </div> |
| <div class="line"><a name="l03262"></a><span class="lineno"> 3262</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l03263"></a><span class="lineno"> 3263</span>  <span class="comment">/*</span></div> |
| <div class="line"><a name="l03264"></a><span class="lineno"> 3264</span> <span class="comment"> * if the node (whose id is "jump") doesn't exist, </span></div> |
| <div class="line"><a name="l03265"></a><span class="lineno"> 3265</span> <span class="comment"> * then insert them into result table </span></div> |
| <div class="line"><a name="l03266"></a><span class="lineno"> 3266</span> <span class="comment"> * (be classified to max_class of its corrsponding node)</span></div> |
| <div class="line"><a name="l03267"></a><span class="lineno"> 3267</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l03268"></a><span class="lineno"> 3268</span>  FOR tree_id IN EXECUTE 'SELECT DISTINCT tid FROM '||tree_table_name LOOP</div> |
| <div class="line"><a name="l03269"></a><span class="lineno"> 3269</span>  SELECT MADLIB_SCHEMA.__format(</div> |
| <div class="line"><a name="l03270"></a><span class="lineno"> 3270</span>  'INSERT INTO %(tid,id, jump, class, prob, parent_id, leaf_id) </div> |
| <div class="line"><a name="l03271"></a><span class="lineno"> 3271</span>  SELECT tid,id, 0, class, prob, parent_id, leaf_id</div> |
| <div class="line"><a name="l03272"></a><span class="lineno"> 3272</span>  FROM %</div> |
| <div class="line"><a name="l03273"></a><span class="lineno"> 3273</span>  WHERE jump NOT IN (SELECT id FROM % WHERE tid=%) </div> |
| <div class="line"><a name="l03274"></a><span class="lineno"> 3274</span>  AND tid=%',</div> |
| <div class="line"><a name="l03275"></a><span class="lineno"> 3275</span>  ARRAY[</div> |
| <div class="line"><a name="l03276"></a><span class="lineno"> 3276</span>  result_table_name,</div> |
| <div class="line"><a name="l03277"></a><span class="lineno"> 3277</span>  table_names[table_pick],</div> |
| <div class="line"><a name="l03278"></a><span class="lineno"> 3278</span>  tree_table_name,</div> |
| <div class="line"><a name="l03279"></a><span class="lineno"> 3279</span>  MADLIB_SCHEMA.__to_char(tree_id),</div> |
| <div class="line"><a name="l03280"></a><span class="lineno"> 3280</span>  MADLIB_SCHEMA.__to_char(tree_id)</div> |
| <div class="line"><a name="l03281"></a><span class="lineno"> 3281</span>  ]</div> |
| <div class="line"><a name="l03282"></a><span class="lineno"> 3282</span>  ) </div> |
| <div class="line"><a name="l03283"></a><span class="lineno"> 3283</span>  INTO curstmt;</div> |
| <div class="line"><a name="l03284"></a><span class="lineno"> 3284</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l03285"></a><span class="lineno"> 3285</span>  </div> |
| <div class="line"><a name="l03286"></a><span class="lineno"> 3286</span>  -- delete from the being classified data table</div> |
| <div class="line"><a name="l03287"></a><span class="lineno"> 3287</span>  SELECT MADLIB_SCHEMA.__format(</div> |
| <div class="line"><a name="l03288"></a><span class="lineno"> 3288</span>  'DELETE FROM %</div> |
| <div class="line"><a name="l03289"></a><span class="lineno"> 3289</span>  WHERE jump NOT IN (SELECT id FROM % WHERE tid=%)</div> |
| <div class="line"><a name="l03290"></a><span class="lineno"> 3290</span>  AND tid=%',</div> |
| <div class="line"><a name="l03291"></a><span class="lineno"> 3291</span>  ARRAY[</div> |
| <div class="line"><a name="l03292"></a><span class="lineno"> 3292</span>  table_names[table_pick],</div> |
| <div class="line"><a name="l03293"></a><span class="lineno"> 3293</span>  tree_table_name,</div> |
| <div class="line"><a name="l03294"></a><span class="lineno"> 3294</span>  MADLIB_SCHEMA.__to_char(tree_id),</div> |
| <div class="line"><a name="l03295"></a><span class="lineno"> 3295</span>  MADLIB_SCHEMA.__to_char(tree_id)</div> |
| <div class="line"><a name="l03296"></a><span class="lineno"> 3296</span>  ]</div> |
| <div class="line"><a name="l03297"></a><span class="lineno"> 3297</span>  ) </div> |
| <div class="line"><a name="l03298"></a><span class="lineno"> 3298</span>  INTO curstmt;</div> |
| <div class="line"><a name="l03299"></a><span class="lineno"> 3299</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l03300"></a><span class="lineno"> 3300</span>  END LOOP; </div> |
| <div class="line"><a name="l03301"></a><span class="lineno"> 3301</span>  END LOOP;</div> |
| <div class="line"><a name="l03302"></a><span class="lineno"> 3302</span> </div> |
| <div class="line"><a name="l03303"></a><span class="lineno"> 3303</span>  EXECUTE 'INSERT INTO '||result_table_name||' SELECT * FROM '|| </div> |
| <div class="line"><a name="l03304"></a><span class="lineno"> 3304</span>  table_names[table_pick] ||' WHERE jump = 0;';</div> |
| <div class="line"><a name="l03305"></a><span class="lineno"> 3305</span>  EXECUTE 'INSERT INTO '||result_table_name||' SELECT * FROM '|| </div> |
| <div class="line"><a name="l03306"></a><span class="lineno"> 3306</span>  table_names[table_pick % 2 + 1] ||' WHERE jump = 0;';</div> |
| <div class="line"><a name="l03307"></a><span class="lineno"> 3307</span>  </div> |
| <div class="line"><a name="l03308"></a><span class="lineno"> 3308</span>  IF (verbosity > 0) THEN </div> |
| <div class="line"><a name="l03309"></a><span class="lineno"> 3309</span>  RAISE INFO 'final classification time:%', clock_timestamp() - time_stamp;</div> |
| <div class="line"><a name="l03310"></a><span class="lineno"> 3310</span>  END IF;</div> |
| <div class="line"><a name="l03311"></a><span class="lineno"> 3311</span>  </div> |
| <div class="line"><a name="l03312"></a><span class="lineno"> 3312</span>  RETURN ARRAY[encoded_table_name, result_table_name];</div> |
| <div class="line"><a name="l03313"></a><span class="lineno"> 3313</span> END</div> |
| <div class="line"><a name="l03314"></a><span class="lineno"> 3314</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l03315"></a><span class="lineno"> 3315</span> </div> |
| <div class="line"><a name="l03316"></a><span class="lineno"> 3316</span> </div> |
| <div class="line"><a name="l03317"></a><span class="lineno"> 3317</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l03318"></a><span class="lineno"> 3318</span> <span class="comment"> * @brief An internal classification function. It classifies with one tree </span></div> |
| <div class="line"><a name="l03319"></a><span class="lineno"> 3319</span> <span class="comment"> * after another. For large data sets, tests shows that it is more</span></div> |
| <div class="line"><a name="l03320"></a><span class="lineno"> 3320</span> <span class="comment"> * efficient than the parallel classification function. </span></div> |
| <div class="line"><a name="l03321"></a><span class="lineno"> 3321</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03322"></a><span class="lineno"> 3322</span> <span class="comment"> * @param classification_table_name The full name of the table containing the </span></div> |
| <div class="line"><a name="l03323"></a><span class="lineno"> 3323</span> <span class="comment"> * classification set.</span></div> |
| <div class="line"><a name="l03324"></a><span class="lineno"> 3324</span> <span class="comment"> * @param tree_table_name The full name of the tree table.</span></div> |
| <div class="line"><a name="l03325"></a><span class="lineno"> 3325</span> <span class="comment"> * @param verbosity > 0 means this function runs in verbose mode. </span></div> |
| <div class="line"><a name="l03326"></a><span class="lineno"> 3326</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03327"></a><span class="lineno"> 3327</span> <span class="comment"> * @return An array containing the encoded table name and classification result </span></div> |
| <div class="line"><a name="l03328"></a><span class="lineno"> 3328</span> <span class="comment"> * table name (We encode the source table during the classification).</span></div> |
| <div class="line"><a name="l03329"></a><span class="lineno"> 3329</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03330"></a><span class="lineno"> 3330</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l03331"></a><span class="lineno"> 3331</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__treemodel_classify_internal_serial</div> |
| <div class="line"><a name="l03332"></a><span class="lineno"> 3332</span>  (</div> |
| <div class="line"><a name="l03333"></a><span class="lineno"> 3333</span>  classification_table_name TEXT, </div> |
| <div class="line"><a name="l03334"></a><span class="lineno"> 3334</span>  tree_table_name TEXT, </div> |
| <div class="line"><a name="l03335"></a><span class="lineno"> 3335</span>  verbosity INT</div> |
| <div class="line"><a name="l03336"></a><span class="lineno"> 3336</span>  ) </div> |
| <div class="line"><a name="l03337"></a><span class="lineno"> 3337</span> RETURNS TEXT[] AS $$</div> |
| <div class="line"><a name="l03338"></a><span class="lineno"> 3338</span> DECLARE</div> |
| <div class="line"><a name="l03339"></a><span class="lineno"> 3339</span>  table_pick INT := 1;</div> |
| <div class="line"><a name="l03340"></a><span class="lineno"> 3340</span>  remains_to_classify INT;</div> |
| <div class="line"><a name="l03341"></a><span class="lineno"> 3341</span>  size_finished INT;</div> |
| <div class="line"><a name="l03342"></a><span class="lineno"> 3342</span>  time_stamp TIMESTAMP;</div> |
| <div class="line"><a name="l03343"></a><span class="lineno"> 3343</span>  metatable_name TEXT := '';</div> |
| <div class="line"><a name="l03344"></a><span class="lineno"> 3344</span>  id_col_name TEXT := 'id';</div> |
| <div class="line"><a name="l03345"></a><span class="lineno"> 3345</span>  curr_level INT := 1;</div> |
| <div class="line"><a name="l03346"></a><span class="lineno"> 3346</span>  max_level INT := 0;</div> |
| <div class="line"><a name="l03347"></a><span class="lineno"> 3347</span>  h2hmv_routine_id INT := 0;</div> |
| <div class="line"><a name="l03348"></a><span class="lineno"> 3348</span>  curstmt TEXT := '';</div> |
| <div class="line"><a name="l03349"></a><span class="lineno"> 3349</span>  result_table_name TEXT := 'dt_classify_internal_rt';</div> |
| <div class="line"><a name="l03350"></a><span class="lineno"> 3350</span>  encoded_table_name TEXT := 'dt_classify_internal_edt';</div> |
| <div class="line"><a name="l03351"></a><span class="lineno"> 3351</span>  table_names TEXT[] := ARRAY[ </div> |
| <div class="line"><a name="l03352"></a><span class="lineno"> 3352</span>  'classified_instance_ping',</div> |
| <div class="line"><a name="l03353"></a><span class="lineno"> 3353</span>  'classified_instance_pong'</div> |
| <div class="line"><a name="l03354"></a><span class="lineno"> 3354</span>  ];</div> |
| <div class="line"><a name="l03355"></a><span class="lineno"> 3355</span>  tree_id INT;</div> |
| <div class="line"><a name="l03356"></a><span class="lineno"> 3356</span>  root_id INT;</div> |
| <div class="line"><a name="l03357"></a><span class="lineno"> 3357</span> BEGIN</div> |
| <div class="line"><a name="l03358"></a><span class="lineno"> 3358</span>  time_stamp = clock_timestamp();</div> |
| <div class="line"><a name="l03359"></a><span class="lineno"> 3359</span> </div> |
| <div class="line"><a name="l03360"></a><span class="lineno"> 3360</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03361"></a><span class="lineno"> 3361</span>  (</div> |
| <div class="line"><a name="l03362"></a><span class="lineno"> 3362</span>  (classification_table_name IS NOT NULL) AND</div> |
| <div class="line"><a name="l03363"></a><span class="lineno"> 3363</span>  (</div> |
| <div class="line"><a name="l03364"></a><span class="lineno"> 3364</span>  MADLIB_SCHEMA.__table_exists</div> |
| <div class="line"><a name="l03365"></a><span class="lineno"> 3365</span>  (</div> |
| <div class="line"><a name="l03366"></a><span class="lineno"> 3366</span>  classification_table_name</div> |
| <div class="line"><a name="l03367"></a><span class="lineno"> 3367</span>  )</div> |
| <div class="line"><a name="l03368"></a><span class="lineno"> 3368</span>  ),</div> |
| <div class="line"><a name="l03369"></a><span class="lineno"> 3369</span>  'the specified classification table' || </div> |
| <div class="line"><a name="l03370"></a><span class="lineno"> 3370</span>  coalesce('<' || </div> |
| <div class="line"><a name="l03371"></a><span class="lineno"> 3371</span>  classification_table_name || </div> |
| <div class="line"><a name="l03372"></a><span class="lineno"> 3372</span>  '> does not exists', ' is NULL')</div> |
| <div class="line"><a name="l03373"></a><span class="lineno"> 3373</span>  ); </div> |
| <div class="line"><a name="l03374"></a><span class="lineno"> 3374</span> </div> |
| <div class="line"><a name="l03375"></a><span class="lineno"> 3375</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03376"></a><span class="lineno"> 3376</span>  (</div> |
| <div class="line"><a name="l03377"></a><span class="lineno"> 3377</span>  (tree_table_name IS NOT NULL) AND</div> |
| <div class="line"><a name="l03378"></a><span class="lineno"> 3378</span>  (</div> |
| <div class="line"><a name="l03379"></a><span class="lineno"> 3379</span>  MADLIB_SCHEMA.__table_exists</div> |
| <div class="line"><a name="l03380"></a><span class="lineno"> 3380</span>  (</div> |
| <div class="line"><a name="l03381"></a><span class="lineno"> 3381</span>  tree_table_name</div> |
| <div class="line"><a name="l03382"></a><span class="lineno"> 3382</span>  )</div> |
| <div class="line"><a name="l03383"></a><span class="lineno"> 3383</span>  ),</div> |
| <div class="line"><a name="l03384"></a><span class="lineno"> 3384</span>  'the specified tree table' || </div> |
| <div class="line"><a name="l03385"></a><span class="lineno"> 3385</span>  coalesce('<' || </div> |
| <div class="line"><a name="l03386"></a><span class="lineno"> 3386</span>  tree_table_name || </div> |
| <div class="line"><a name="l03387"></a><span class="lineno"> 3387</span>  '> does not exists', ' is NULL')</div> |
| <div class="line"><a name="l03388"></a><span class="lineno"> 3388</span>  ); </div> |
| <div class="line"><a name="l03389"></a><span class="lineno"> 3389</span> </div> |
| <div class="line"><a name="l03390"></a><span class="lineno"> 3390</span>  </div> |
| <div class="line"><a name="l03391"></a><span class="lineno"> 3391</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03392"></a><span class="lineno"> 3392</span>  (</div> |
| <div class="line"><a name="l03393"></a><span class="lineno"> 3393</span>  verbosity IS NOT NULL, </div> |
| <div class="line"><a name="l03394"></a><span class="lineno"> 3394</span>  'verbosity must be non-null'</div> |
| <div class="line"><a name="l03395"></a><span class="lineno"> 3395</span>  ); </div> |
| <div class="line"><a name="l03396"></a><span class="lineno"> 3396</span>  </div> |
| <div class="line"><a name="l03397"></a><span class="lineno"> 3397</span>  EXECUTE 'DROP TABLE IF EXISTS ' || encoded_table_name || ' CASCADE';</div> |
| <div class="line"><a name="l03398"></a><span class="lineno"> 3398</span>  </div> |
| <div class="line"><a name="l03399"></a><span class="lineno"> 3399</span>  metatable_name = MADLIB_SCHEMA.__get_metatable_name(tree_table_name);</div> |
| <div class="line"><a name="l03400"></a><span class="lineno"> 3400</span> </div> |
| <div class="line"><a name="l03401"></a><span class="lineno"> 3401</span>  h2hmv_routine_id = MADLIB_SCHEMA.__get_routine_id(tree_table_name);</div> |
| <div class="line"><a name="l03402"></a><span class="lineno"> 3402</span>  </div> |
| <div class="line"><a name="l03403"></a><span class="lineno"> 3403</span>  PERFORM MADLIB_SCHEMA.__encode_table</div> |
| <div class="line"><a name="l03404"></a><span class="lineno"> 3404</span>  (</div> |
| <div class="line"><a name="l03405"></a><span class="lineno"> 3405</span>  classification_table_name, </div> |
| <div class="line"><a name="l03406"></a><span class="lineno"> 3406</span>  encoded_table_name, </div> |
| <div class="line"><a name="l03407"></a><span class="lineno"> 3407</span>  metatable_name, </div> |
| <div class="line"><a name="l03408"></a><span class="lineno"> 3408</span>  h2hmv_routine_id,</div> |
| <div class="line"><a name="l03409"></a><span class="lineno"> 3409</span>  verbosity</div> |
| <div class="line"><a name="l03410"></a><span class="lineno"> 3410</span>  );</div> |
| <div class="line"><a name="l03411"></a><span class="lineno"> 3411</span>  </div> |
| <div class="line"><a name="l03412"></a><span class="lineno"> 3412</span>  IF (verbosity > 0) THEN</div> |
| <div class="line"><a name="l03413"></a><span class="lineno"> 3413</span>  RAISE INFO 'tabular format. id_col_name: %', id_col_name;</div> |
| <div class="line"><a name="l03414"></a><span class="lineno"> 3414</span>  END IF; </div> |
| <div class="line"><a name="l03415"></a><span class="lineno"> 3415</span>  </div> |
| <div class="line"><a name="l03416"></a><span class="lineno"> 3416</span>  <span class="comment">/*</span></div> |
| <div class="line"><a name="l03417"></a><span class="lineno"> 3417</span> <span class="comment"> * The table of classified_instance_ping and classified_instance_pong are</span></div> |
| <div class="line"><a name="l03418"></a><span class="lineno"> 3418</span> <span class="comment"> * auxiliary tables used during the classification process.</span></div> |
| <div class="line"><a name="l03419"></a><span class="lineno"> 3419</span> <span class="comment"> * For each record, these tables tell us which node it belongs to. They also</span></div> |
| <div class="line"><a name="l03420"></a><span class="lineno"> 3420</span> <span class="comment"> * hold the information of class and probability.</span></div> |
| <div class="line"><a name="l03421"></a><span class="lineno"> 3421</span> <span class="comment"> * We use transfer data between these two tables rather than update a single</span></div> |
| <div class="line"><a name="l03422"></a><span class="lineno"> 3422</span> <span class="comment"> * table during the classification process. We find the operation of update</span></div> |
| <div class="line"><a name="l03423"></a><span class="lineno"> 3423</span> <span class="comment"> * is quite expensive.</span></div> |
| <div class="line"><a name="l03424"></a><span class="lineno"> 3424</span> <span class="comment"> */</span> </div> |
| <div class="line"><a name="l03425"></a><span class="lineno"> 3425</span>  DROP TABLE IF EXISTS classified_instance_ping;</div> |
| <div class="line"><a name="l03426"></a><span class="lineno"> 3426</span>  CREATE TEMP TABLE classified_instance_ping</div> |
| <div class="line"><a name="l03427"></a><span class="lineno"> 3427</span>  (</div> |
| <div class="line"><a name="l03428"></a><span class="lineno"> 3428</span>  id BIGINT,</div> |
| <div class="line"><a name="l03429"></a><span class="lineno"> 3429</span>  jump INT,</div> |
| <div class="line"><a name="l03430"></a><span class="lineno"> 3430</span>  class INT,</div> |
| <div class="line"><a name="l03431"></a><span class="lineno"> 3431</span>  prob FLOAT,</div> |
| <div class="line"><a name="l03432"></a><span class="lineno"> 3432</span>  parent_id INT,</div> |
| <div class="line"><a name="l03433"></a><span class="lineno"> 3433</span>  leaf_id INT</div> |
| <div class="line"><a name="l03434"></a><span class="lineno"> 3434</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (id)');</div> |
| <div class="line"><a name="l03435"></a><span class="lineno"> 3435</span>  </div> |
| <div class="line"><a name="l03436"></a><span class="lineno"> 3436</span>  DROP TABLE IF EXISTS classified_instance_pong;</div> |
| <div class="line"><a name="l03437"></a><span class="lineno"> 3437</span>  CREATE TEMP TABLE classified_instance_pong</div> |
| <div class="line"><a name="l03438"></a><span class="lineno"> 3438</span>  (</div> |
| <div class="line"><a name="l03439"></a><span class="lineno"> 3439</span>  id BIGINT,</div> |
| <div class="line"><a name="l03440"></a><span class="lineno"> 3440</span>  jump INT,</div> |
| <div class="line"><a name="l03441"></a><span class="lineno"> 3441</span>  class INT,</div> |
| <div class="line"><a name="l03442"></a><span class="lineno"> 3442</span>  prob FLOAT,</div> |
| <div class="line"><a name="l03443"></a><span class="lineno"> 3443</span>  parent_id INT,</div> |
| <div class="line"><a name="l03444"></a><span class="lineno"> 3444</span>  leaf_id INT</div> |
| <div class="line"><a name="l03445"></a><span class="lineno"> 3445</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (id)');</div> |
| <div class="line"><a name="l03446"></a><span class="lineno"> 3446</span> </div> |
| <div class="line"><a name="l03447"></a><span class="lineno"> 3447</span>  </div> |
| <div class="line"><a name="l03448"></a><span class="lineno"> 3448</span>  EXECUTE 'DROP TABLE IF EXISTS '||result_table_name || ' CASCADE'; </div> |
| <div class="line"><a name="l03449"></a><span class="lineno"> 3449</span>  EXECUTE 'CREATE TEMP TABLE ' || result_table_name || E'</div> |
| <div class="line"><a name="l03450"></a><span class="lineno"> 3450</span>  (</div> |
| <div class="line"><a name="l03451"></a><span class="lineno"> 3451</span>  tid INT,</div> |
| <div class="line"><a name="l03452"></a><span class="lineno"> 3452</span>  id BIGINT,</div> |
| <div class="line"><a name="l03453"></a><span class="lineno"> 3453</span>  jump INT,</div> |
| <div class="line"><a name="l03454"></a><span class="lineno"> 3454</span>  class INT,</div> |
| <div class="line"><a name="l03455"></a><span class="lineno"> 3455</span>  prob FLOAT,</div> |
| <div class="line"><a name="l03456"></a><span class="lineno"> 3456</span>  parent_id INT,</div> |
| <div class="line"><a name="l03457"></a><span class="lineno"> 3457</span>  leaf_id INT</div> |
| <div class="line"><a name="l03458"></a><span class="lineno"> 3458</span>  ) m4_ifdef(`__GREENPLUM__', `DISTRIBUTED BY (id)');';</div> |
| <div class="line"><a name="l03459"></a><span class="lineno"> 3459</span>  </div> |
| <div class="line"><a name="l03460"></a><span class="lineno"> 3460</span>  FOR tree_id IN EXECUTE 'SELECT DISTINCT tid FROM '||tree_table_name LOOP</div> |
| <div class="line"><a name="l03461"></a><span class="lineno"> 3461</span>  EXECUTE 'SELECT max(array_upper(tree_location,1)) FROM '||</div> |
| <div class="line"><a name="l03462"></a><span class="lineno"> 3462</span>  tree_table_name||' WHERE tid='||tree_id||';' INTO max_level;</div> |
| <div class="line"><a name="l03463"></a><span class="lineno"> 3463</span>  IF (verbosity > 0) THEN </div> |
| <div class="line"><a name="l03464"></a><span class="lineno"> 3464</span>  RAISE INFO 'tree_id: %, max_level: %', tree_id,max_level;</div> |
| <div class="line"><a name="l03465"></a><span class="lineno"> 3465</span>  END IF;</div> |
| <div class="line"><a name="l03466"></a><span class="lineno"> 3466</span> </div> |
| <div class="line"><a name="l03467"></a><span class="lineno"> 3467</span> </div> |
| <div class="line"><a name="l03468"></a><span class="lineno"> 3468</span>  IF( max_level is NULL ) THEN</div> |
| <div class="line"><a name="l03469"></a><span class="lineno"> 3469</span>  RAISE EXCEPTION 'tree should not be empty';</div> |
| <div class="line"><a name="l03470"></a><span class="lineno"> 3470</span>  END IF;</div> |
| <div class="line"><a name="l03471"></a><span class="lineno"> 3471</span>  </div> |
| <div class="line"><a name="l03472"></a><span class="lineno"> 3472</span>  TRUNCATE classified_instance_ping;</div> |
| <div class="line"><a name="l03473"></a><span class="lineno"> 3473</span>  TRUNCATE classified_instance_pong;</div> |
| <div class="line"><a name="l03474"></a><span class="lineno"> 3474</span> </div> |
| <div class="line"><a name="l03475"></a><span class="lineno"> 3475</span>  EXECUTE 'SELECT id FROM '||tree_table_name||</div> |
| <div class="line"><a name="l03476"></a><span class="lineno"> 3476</span>  ' WHERE parent_id=0 and tid='||tree_id||';' INTO root_id;</div> |
| <div class="line"><a name="l03477"></a><span class="lineno"> 3477</span>  EXECUTE 'INSERT INTO classified_instance_ping (id, jump, class, prob) </div> |
| <div class="line"><a name="l03478"></a><span class="lineno"> 3478</span>  SELECT '||id_col_name||', '||root_id||', 0, 0 FROM ' || </div> |
| <div class="line"><a name="l03479"></a><span class="lineno"> 3479</span>  encoded_table_name || ';'; </div> |
| <div class="line"><a name="l03480"></a><span class="lineno"> 3480</span>  table_pick= 1;</div> |
| <div class="line"><a name="l03481"></a><span class="lineno"> 3481</span>  FOR curr_level IN 1..max_level LOOP</div> |
| <div class="line"><a name="l03482"></a><span class="lineno"> 3482</span>  IF (verbosity > 0) THEN </div> |
| <div class="line"><a name="l03483"></a><span class="lineno"> 3483</span>  RAISE INFO 'new_depth: %', curr_level;</div> |
| <div class="line"><a name="l03484"></a><span class="lineno"> 3484</span>  END IF;</div> |
| <div class="line"><a name="l03485"></a><span class="lineno"> 3485</span>  </div> |
| <div class="line"><a name="l03486"></a><span class="lineno"> 3486</span>  table_pick = table_pick % 2 + 1; </div> |
| <div class="line"><a name="l03487"></a><span class="lineno"> 3487</span>  </div> |
| <div class="line"><a name="l03488"></a><span class="lineno"> 3488</span>  EXECUTE 'TRUNCATE '|| table_names[table_pick] ||';';</div> |
| <div class="line"><a name="l03489"></a><span class="lineno"> 3489</span>  EXECUTE 'SELECT count(id) FROM '||result_table_name||';' </div> |
| <div class="line"><a name="l03490"></a><span class="lineno"> 3490</span>  INTO size_finished;</div> |
| <div class="line"><a name="l03491"></a><span class="lineno"> 3491</span>  </div> |
| <div class="line"><a name="l03492"></a><span class="lineno"> 3492</span>  IF (verbosity > 0) THEN </div> |
| <div class="line"><a name="l03493"></a><span class="lineno"> 3493</span>  RAISE INFO 'size_finished %', size_finished;</div> |
| <div class="line"><a name="l03494"></a><span class="lineno"> 3494</span>  END IF; </div> |
| <div class="line"><a name="l03495"></a><span class="lineno"> 3495</span>  </div> |
| <div class="line"><a name="l03496"></a><span class="lineno"> 3496</span>  EXECUTE 'SELECT count(*) FROM '|| </div> |
| <div class="line"><a name="l03497"></a><span class="lineno"> 3497</span>  table_names[(table_pick) % 2 + 1] ||';' </div> |
| <div class="line"><a name="l03498"></a><span class="lineno"> 3498</span>  INTO remains_to_classify;</div> |
| <div class="line"><a name="l03499"></a><span class="lineno"> 3499</span>  </div> |
| <div class="line"><a name="l03500"></a><span class="lineno"> 3500</span>  IF (remains_to_classify = 0) THEN</div> |
| <div class="line"><a name="l03501"></a><span class="lineno"> 3501</span>  IF (verbosity > 0) THEN </div> |
| <div class="line"><a name="l03502"></a><span class="lineno"> 3502</span>  RAISE INFO 'size_finished: % remains_to_classify: %', </div> |
| <div class="line"><a name="l03503"></a><span class="lineno"> 3503</span>  size_finished, remains_to_classify;</div> |
| <div class="line"><a name="l03504"></a><span class="lineno"> 3504</span>  END IF; </div> |
| <div class="line"><a name="l03505"></a><span class="lineno"> 3505</span>  </div> |
| <div class="line"><a name="l03506"></a><span class="lineno"> 3506</span>  EXIT;</div> |
| <div class="line"><a name="l03507"></a><span class="lineno"> 3507</span>  END IF;</div> |
| <div class="line"><a name="l03508"></a><span class="lineno"> 3508</span> </div> |
| <div class="line"><a name="l03509"></a><span class="lineno"> 3509</span>  SELECT MADLIB_SCHEMA.__format(</div> |
| <div class="line"><a name="l03510"></a><span class="lineno"> 3510</span>  'INSERT INTO %</div> |
| <div class="line"><a name="l03511"></a><span class="lineno"> 3511</span>  SELECT pt.id, </div> |
| <div class="line"><a name="l03512"></a><span class="lineno"> 3512</span>  CASE WHEN (is_cont) THEN </div> |
| <div class="line"><a name="l03513"></a><span class="lineno"> 3513</span>  CASE WHEN (gt.lmc_nid IS NULL) THEN</div> |
| <div class="line"><a name="l03514"></a><span class="lineno"> 3514</span>  0</div> |
| <div class="line"><a name="l03515"></a><span class="lineno"> 3515</span>  ELSE</div> |
| <div class="line"><a name="l03516"></a><span class="lineno"> 3516</span>  gt.lmc_nid + </div> |
| <div class="line"><a name="l03517"></a><span class="lineno"> 3517</span>  float8lt(gt.split_value, fvals[gt.feature])::INT4 </div> |
| <div class="line"><a name="l03518"></a><span class="lineno"> 3518</span>  + 1 - gt.lmc_fval</div> |
| <div class="line"><a name="l03519"></a><span class="lineno"> 3519</span>  END</div> |
| <div class="line"><a name="l03520"></a><span class="lineno"> 3520</span>  ELSE </div> |
| <div class="line"><a name="l03521"></a><span class="lineno"> 3521</span>  CASE WHEN (gt.lmc_nid IS NULL) THEN</div> |
| <div class="line"><a name="l03522"></a><span class="lineno"> 3522</span>  0</div> |
| <div class="line"><a name="l03523"></a><span class="lineno"> 3523</span>  ELSE </div> |
| <div class="line"><a name="l03524"></a><span class="lineno"> 3524</span>  gt.lmc_nid + fvals[gt.feature] - gt.lmc_fval</div> |
| <div class="line"><a name="l03525"></a><span class="lineno"> 3525</span>  END</div> |
| <div class="line"><a name="l03526"></a><span class="lineno"> 3526</span>  END as newjump,</div> |
| <div class="line"><a name="l03527"></a><span class="lineno"> 3527</span>  gt.max_class, gt.probability, gt.parent_id, gt.id </div> |
| <div class="line"><a name="l03528"></a><span class="lineno"> 3528</span>  FROM </div> |
| <div class="line"><a name="l03529"></a><span class="lineno"> 3529</span>  (</div> |
| <div class="line"><a name="l03530"></a><span class="lineno"> 3530</span>  SELECT t1.id, t1.jump, fvals </div> |
| <div class="line"><a name="l03531"></a><span class="lineno"> 3531</span>  FROM % t1 </div> |
| <div class="line"><a name="l03532"></a><span class="lineno"> 3532</span>  LEFT JOIN % t2 </div> |
| <div class="line"><a name="l03533"></a><span class="lineno"> 3533</span>  ON t1.id = t2.id</div> |
| <div class="line"><a name="l03534"></a><span class="lineno"> 3534</span>  ) AS pt,</div> |
| <div class="line"><a name="l03535"></a><span class="lineno"> 3535</span>  (</div> |
| <div class="line"><a name="l03536"></a><span class="lineno"> 3536</span>  SELECT lmc_nid, lmc_fval, max_class, feature, probability, </div> |
| <div class="line"><a name="l03537"></a><span class="lineno"> 3537</span>  parent_id, id, is_cont, split_value</div> |
| <div class="line"><a name="l03538"></a><span class="lineno"> 3538</span>  FROM % </div> |
| <div class="line"><a name="l03539"></a><span class="lineno"> 3539</span>  WHERE array_upper(tree_location,1) = % AND tid=%</div> |
| <div class="line"><a name="l03540"></a><span class="lineno"> 3540</span>  ) AS gt</div> |
| <div class="line"><a name="l03541"></a><span class="lineno"> 3541</span>  WHERE pt.jump = gt.id;',</div> |
| <div class="line"><a name="l03542"></a><span class="lineno"> 3542</span>  ARRAY[</div> |
| <div class="line"><a name="l03543"></a><span class="lineno"> 3543</span>  table_names[table_pick],</div> |
| <div class="line"><a name="l03544"></a><span class="lineno"> 3544</span>  table_names[(table_pick) % 2 + 1],</div> |
| <div class="line"><a name="l03545"></a><span class="lineno"> 3545</span>  encoded_table_name,</div> |
| <div class="line"><a name="l03546"></a><span class="lineno"> 3546</span>  tree_table_name,</div> |
| <div class="line"><a name="l03547"></a><span class="lineno"> 3547</span>  MADLIB_SCHEMA.__to_char(curr_level),</div> |
| <div class="line"><a name="l03548"></a><span class="lineno"> 3548</span>  MADLIB_SCHEMA.__to_char(tree_id)</div> |
| <div class="line"><a name="l03549"></a><span class="lineno"> 3549</span>  ]</div> |
| <div class="line"><a name="l03550"></a><span class="lineno"> 3550</span>  )</div> |
| <div class="line"><a name="l03551"></a><span class="lineno"> 3551</span>  INTO curstmt; </div> |
| <div class="line"><a name="l03552"></a><span class="lineno"> 3552</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l03553"></a><span class="lineno"> 3553</span>  </div> |
| <div class="line"><a name="l03554"></a><span class="lineno"> 3554</span>  <span class="comment">/*</span></div> |
| <div class="line"><a name="l03555"></a><span class="lineno"> 3555</span> <span class="comment"> * if the node (whose id is "jump") doesn't exist, </span></div> |
| <div class="line"><a name="l03556"></a><span class="lineno"> 3556</span> <span class="comment"> * then insert them into result table </span></div> |
| <div class="line"><a name="l03557"></a><span class="lineno"> 3557</span> <span class="comment"> * (be classified to max_class of its corrsponding node)</span></div> |
| <div class="line"><a name="l03558"></a><span class="lineno"> 3558</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l03559"></a><span class="lineno"> 3559</span>  SELECT MADLIB_SCHEMA.__format(</div> |
| <div class="line"><a name="l03560"></a><span class="lineno"> 3560</span>  'INSERT INTO %(tid,id, jump, class, prob, parent_id, leaf_id) </div> |
| <div class="line"><a name="l03561"></a><span class="lineno"> 3561</span>  SELECT '||tree_id||',id, 0, class, prob, parent_id, leaf_id</div> |
| <div class="line"><a name="l03562"></a><span class="lineno"> 3562</span>  FROM %</div> |
| <div class="line"><a name="l03563"></a><span class="lineno"> 3563</span>  WHERE jump NOT IN (SELECT id FROM % WHERE tid=%)',</div> |
| <div class="line"><a name="l03564"></a><span class="lineno"> 3564</span>  ARRAY[</div> |
| <div class="line"><a name="l03565"></a><span class="lineno"> 3565</span>  result_table_name,</div> |
| <div class="line"><a name="l03566"></a><span class="lineno"> 3566</span>  table_names[table_pick],</div> |
| <div class="line"><a name="l03567"></a><span class="lineno"> 3567</span>  tree_table_name,</div> |
| <div class="line"><a name="l03568"></a><span class="lineno"> 3568</span>  MADLIB_SCHEMA.__to_char(tree_id)</div> |
| <div class="line"><a name="l03569"></a><span class="lineno"> 3569</span>  ]</div> |
| <div class="line"><a name="l03570"></a><span class="lineno"> 3570</span>  ) </div> |
| <div class="line"><a name="l03571"></a><span class="lineno"> 3571</span>  INTO curstmt;</div> |
| <div class="line"><a name="l03572"></a><span class="lineno"> 3572</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l03573"></a><span class="lineno"> 3573</span>  </div> |
| <div class="line"><a name="l03574"></a><span class="lineno"> 3574</span>  -- delete from the being classified data table</div> |
| <div class="line"><a name="l03575"></a><span class="lineno"> 3575</span>  SELECT MADLIB_SCHEMA.__format(</div> |
| <div class="line"><a name="l03576"></a><span class="lineno"> 3576</span>  'DELETE FROM %</div> |
| <div class="line"><a name="l03577"></a><span class="lineno"> 3577</span>  WHERE jump NOT IN (SELECT id FROM % WHERE tid=%)',</div> |
| <div class="line"><a name="l03578"></a><span class="lineno"> 3578</span>  ARRAY[</div> |
| <div class="line"><a name="l03579"></a><span class="lineno"> 3579</span>  table_names[table_pick],</div> |
| <div class="line"><a name="l03580"></a><span class="lineno"> 3580</span>  tree_table_name,</div> |
| <div class="line"><a name="l03581"></a><span class="lineno"> 3581</span>  MADLIB_SCHEMA.__to_char(tree_id)</div> |
| <div class="line"><a name="l03582"></a><span class="lineno"> 3582</span>  ]</div> |
| <div class="line"><a name="l03583"></a><span class="lineno"> 3583</span>  ) </div> |
| <div class="line"><a name="l03584"></a><span class="lineno"> 3584</span>  INTO curstmt;</div> |
| <div class="line"><a name="l03585"></a><span class="lineno"> 3585</span>  EXECUTE curstmt;</div> |
| <div class="line"><a name="l03586"></a><span class="lineno"> 3586</span>  END LOOP;</div> |
| <div class="line"><a name="l03587"></a><span class="lineno"> 3587</span> </div> |
| <div class="line"><a name="l03588"></a><span class="lineno"> 3588</span>  EXECUTE 'INSERT INTO '||result_table_name||' SELECT '||tree_id||',* FROM '|| </div> |
| <div class="line"><a name="l03589"></a><span class="lineno"> 3589</span>  table_names[table_pick] ||' WHERE jump = 0;';</div> |
| <div class="line"><a name="l03590"></a><span class="lineno"> 3590</span>  EXECUTE 'INSERT INTO '||result_table_name||' SELECT '||tree_id||',* FROM '|| </div> |
| <div class="line"><a name="l03591"></a><span class="lineno"> 3591</span>  table_names[table_pick % 2 + 1] ||' WHERE jump = 0;';</div> |
| <div class="line"><a name="l03592"></a><span class="lineno"> 3592</span>  END LOOP;</div> |
| <div class="line"><a name="l03593"></a><span class="lineno"> 3593</span> </div> |
| <div class="line"><a name="l03594"></a><span class="lineno"> 3594</span>  IF (verbosity > 0) THEN </div> |
| <div class="line"><a name="l03595"></a><span class="lineno"> 3595</span>  RAISE INFO 'final classification time:%', clock_timestamp() - time_stamp;</div> |
| <div class="line"><a name="l03596"></a><span class="lineno"> 3596</span>  END IF;</div> |
| <div class="line"><a name="l03597"></a><span class="lineno"> 3597</span>  </div> |
| <div class="line"><a name="l03598"></a><span class="lineno"> 3598</span>  RETURN ARRAY[encoded_table_name, result_table_name];</div> |
| <div class="line"><a name="l03599"></a><span class="lineno"> 3599</span> END</div> |
| <div class="line"><a name="l03600"></a><span class="lineno"> 3600</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l03601"></a><span class="lineno"> 3601</span> </div> |
| <div class="line"><a name="l03602"></a><span class="lineno"> 3602</span> </div> |
| <div class="line"><a name="l03603"></a><span class="lineno"> 3603</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l03604"></a><span class="lineno"> 3604</span> <span class="comment"> * @brief This function check the accuracy of the trained tree model.</span></div> |
| <div class="line"><a name="l03605"></a><span class="lineno"> 3605</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l03606"></a><span class="lineno"> 3606</span> <span class="comment"> * @param tree_table_name The name of the tree containing the model.</span></div> |
| <div class="line"><a name="l03607"></a><span class="lineno"> 3607</span> <span class="comment"> * @param scoring_table_name The full name of the table/view with the </span></div> |
| <div class="line"><a name="l03608"></a><span class="lineno"> 3608</span> <span class="comment"> * data to be scored.</span></div> |
| <div class="line"><a name="l03609"></a><span class="lineno"> 3609</span> <span class="comment"> * @param verbosity > 0 means this function runs in verbose mode.</span></div> |
| <div class="line"><a name="l03610"></a><span class="lineno"> 3610</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03611"></a><span class="lineno"> 3611</span> <span class="comment"> * @return The estimated accuracy information.</span></div> |
| <div class="line"><a name="l03612"></a><span class="lineno"> 3612</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03613"></a><span class="lineno"> 3613</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l03614"></a><span class="lineno"> 3614</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__treemodel_score</div> |
| <div class="line"><a name="l03615"></a><span class="lineno"> 3615</span>  (</div> |
| <div class="line"><a name="l03616"></a><span class="lineno"> 3616</span>  tree_table_name TEXT, </div> |
| <div class="line"><a name="l03617"></a><span class="lineno"> 3617</span>  scoring_table_name TEXT, </div> |
| <div class="line"><a name="l03618"></a><span class="lineno"> 3618</span>  verbosity INT</div> |
| <div class="line"><a name="l03619"></a><span class="lineno"> 3619</span>  ) </div> |
| <div class="line"><a name="l03620"></a><span class="lineno"> 3620</span> RETURNS FLOAT AS $$</div> |
| <div class="line"><a name="l03621"></a><span class="lineno"> 3621</span> DECLARE</div> |
| <div class="line"><a name="l03622"></a><span class="lineno"> 3622</span>  result_table_name TEXT;</div> |
| <div class="line"><a name="l03623"></a><span class="lineno"> 3623</span>  result_table_name_final TEXT;</div> |
| <div class="line"><a name="l03624"></a><span class="lineno"> 3624</span>  id_col_name TEXT := 'id';</div> |
| <div class="line"><a name="l03625"></a><span class="lineno"> 3625</span>  class_col_name TEXT := 'class';</div> |
| <div class="line"><a name="l03626"></a><span class="lineno"> 3626</span>  curstmt TEXT := '';</div> |
| <div class="line"><a name="l03627"></a><span class="lineno"> 3627</span>  num_of_row FLOAT := 0.0;</div> |
| <div class="line"><a name="l03628"></a><span class="lineno"> 3628</span>  mis_of_row FLOAT := 0.0;</div> |
| <div class="line"><a name="l03629"></a><span class="lineno"> 3629</span>  encoded_table_name TEXT := '';</div> |
| <div class="line"><a name="l03630"></a><span class="lineno"> 3630</span>  table_names TEXT[];</div> |
| <div class="line"><a name="l03631"></a><span class="lineno"> 3631</span> BEGIN</div> |
| <div class="line"><a name="l03632"></a><span class="lineno"> 3632</span> </div> |
| <div class="line"><a name="l03633"></a><span class="lineno"> 3633</span>  IF (verbosity > 0) THEN</div> |
| <div class="line"><a name="l03634"></a><span class="lineno"> 3634</span>  -- get rid of the messages whose severity level is lower than 'WARNING'</div> |
| <div class="line"><a name="l03635"></a><span class="lineno"> 3635</span>  SET client_min_messages = WARNING;</div> |
| <div class="line"><a name="l03636"></a><span class="lineno"> 3636</span>  END IF;</div> |
| <div class="line"><a name="l03637"></a><span class="lineno"> 3637</span>  </div> |
| <div class="line"><a name="l03638"></a><span class="lineno"> 3638</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03639"></a><span class="lineno"> 3639</span>  (</div> |
| <div class="line"><a name="l03640"></a><span class="lineno"> 3640</span>  (tree_table_name IS NOT NULL) AND</div> |
| <div class="line"><a name="l03641"></a><span class="lineno"> 3641</span>  (</div> |
| <div class="line"><a name="l03642"></a><span class="lineno"> 3642</span>  MADLIB_SCHEMA.__table_exists</div> |
| <div class="line"><a name="l03643"></a><span class="lineno"> 3643</span>  (</div> |
| <div class="line"><a name="l03644"></a><span class="lineno"> 3644</span>  tree_table_name</div> |
| <div class="line"><a name="l03645"></a><span class="lineno"> 3645</span>  )</div> |
| <div class="line"><a name="l03646"></a><span class="lineno"> 3646</span>  ),</div> |
| <div class="line"><a name="l03647"></a><span class="lineno"> 3647</span>  'the specified tree table' || coalesce('<' || tree_table_name </div> |
| <div class="line"><a name="l03648"></a><span class="lineno"> 3648</span>  || '> does not exist', ' is NULL')</div> |
| <div class="line"><a name="l03649"></a><span class="lineno"> 3649</span>  ); </div> |
| <div class="line"><a name="l03650"></a><span class="lineno"> 3650</span> </div> |
| <div class="line"><a name="l03651"></a><span class="lineno"> 3651</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03652"></a><span class="lineno"> 3652</span>  (</div> |
| <div class="line"><a name="l03653"></a><span class="lineno"> 3653</span>  (scoring_table_name IS NOT NULL) AND</div> |
| <div class="line"><a name="l03654"></a><span class="lineno"> 3654</span>  (</div> |
| <div class="line"><a name="l03655"></a><span class="lineno"> 3655</span>  MADLIB_SCHEMA.__table_exists</div> |
| <div class="line"><a name="l03656"></a><span class="lineno"> 3656</span>  (</div> |
| <div class="line"><a name="l03657"></a><span class="lineno"> 3657</span>  scoring_table_name</div> |
| <div class="line"><a name="l03658"></a><span class="lineno"> 3658</span>  )</div> |
| <div class="line"><a name="l03659"></a><span class="lineno"> 3659</span>  ),</div> |
| <div class="line"><a name="l03660"></a><span class="lineno"> 3660</span>  'the specified scoring table' || </div> |
| <div class="line"><a name="l03661"></a><span class="lineno"> 3661</span>  coalesce('<' || scoring_table_name || </div> |
| <div class="line"><a name="l03662"></a><span class="lineno"> 3662</span>  '> does not exist', ' is NULL')</div> |
| <div class="line"><a name="l03663"></a><span class="lineno"> 3663</span>  ); </div> |
| <div class="line"><a name="l03664"></a><span class="lineno"> 3664</span> </div> |
| <div class="line"><a name="l03665"></a><span class="lineno"> 3665</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03666"></a><span class="lineno"> 3666</span>  (</div> |
| <div class="line"><a name="l03667"></a><span class="lineno"> 3667</span>  MADLIB_SCHEMA.__column_exists</div> |
| <div class="line"><a name="l03668"></a><span class="lineno"> 3668</span>  (</div> |
| <div class="line"><a name="l03669"></a><span class="lineno"> 3669</span>  scoring_table_name,</div> |
| <div class="line"><a name="l03670"></a><span class="lineno"> 3670</span>  MADLIB_SCHEMA.__get_class_column_name</div> |
| <div class="line"><a name="l03671"></a><span class="lineno"> 3671</span>  (</div> |
| <div class="line"><a name="l03672"></a><span class="lineno"> 3672</span>  MADLIB_SCHEMA.__get_metatable_name(tree_table_name)</div> |
| <div class="line"><a name="l03673"></a><span class="lineno"> 3673</span>  )</div> |
| <div class="line"><a name="l03674"></a><span class="lineno"> 3674</span>  ),</div> |
| <div class="line"><a name="l03675"></a><span class="lineno"> 3675</span>  'the specified scoring table<' || scoring_table_name || </div> |
| <div class="line"><a name="l03676"></a><span class="lineno"> 3676</span>  '> does not have class column'</div> |
| <div class="line"><a name="l03677"></a><span class="lineno"> 3677</span>  );</div> |
| <div class="line"><a name="l03678"></a><span class="lineno"> 3678</span>  </div> |
| <div class="line"><a name="l03679"></a><span class="lineno"> 3679</span>  table_names = MADLIB_SCHEMA.__treemodel_classify_internal</div> |
| <div class="line"><a name="l03680"></a><span class="lineno"> 3680</span>  (</div> |
| <div class="line"><a name="l03681"></a><span class="lineno"> 3681</span>  scoring_table_name, </div> |
| <div class="line"><a name="l03682"></a><span class="lineno"> 3682</span>  tree_table_name, </div> |
| <div class="line"><a name="l03683"></a><span class="lineno"> 3683</span>  verbosity</div> |
| <div class="line"><a name="l03684"></a><span class="lineno"> 3684</span>  ); </div> |
| <div class="line"><a name="l03685"></a><span class="lineno"> 3685</span>  encoded_table_name = table_names[1];</div> |
| <div class="line"><a name="l03686"></a><span class="lineno"> 3686</span>  result_table_name = table_names[2];</div> |
| <div class="line"><a name="l03687"></a><span class="lineno"> 3687</span>  result_table_name_final = result_table_name||'_final';</div> |
| <div class="line"><a name="l03688"></a><span class="lineno"> 3688</span>  </div> |
| <div class="line"><a name="l03689"></a><span class="lineno"> 3689</span>  PERFORM MADLIB_SCHEMA.__treemodel_get_vote_result</div> |
| <div class="line"><a name="l03690"></a><span class="lineno"> 3690</span>  (</div> |
| <div class="line"><a name="l03691"></a><span class="lineno"> 3691</span>  result_table_name, </div> |
| <div class="line"><a name="l03692"></a><span class="lineno"> 3692</span>  result_table_name_final</div> |
| <div class="line"><a name="l03693"></a><span class="lineno"> 3693</span>  );</div> |
| <div class="line"><a name="l03694"></a><span class="lineno"> 3694</span> </div> |
| <div class="line"><a name="l03695"></a><span class="lineno"> 3695</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l03696"></a><span class="lineno"> 3696</span>  (</div> |
| <div class="line"><a name="l03697"></a><span class="lineno"> 3697</span>  'SELECT count(id) FROM %;',</div> |
| <div class="line"><a name="l03698"></a><span class="lineno"> 3698</span>  result_table_name_final</div> |
| <div class="line"><a name="l03699"></a><span class="lineno"> 3699</span>  ) </div> |
| <div class="line"><a name="l03700"></a><span class="lineno"> 3700</span>  INTO curstmt;</div> |
| <div class="line"><a name="l03701"></a><span class="lineno"> 3701</span>  </div> |
| <div class="line"><a name="l03702"></a><span class="lineno"> 3702</span>  EXECUTE curstmt INTO num_of_row;</div> |
| <div class="line"><a name="l03703"></a><span class="lineno"> 3703</span> </div> |
| <div class="line"><a name="l03704"></a><span class="lineno"> 3704</span>  SELECT MADLIB_SCHEMA.__format</div> |
| <div class="line"><a name="l03705"></a><span class="lineno"> 3705</span>  (</div> |
| <div class="line"><a name="l03706"></a><span class="lineno"> 3706</span>  'SELECT count(t2.id) </div> |
| <div class="line"><a name="l03707"></a><span class="lineno"> 3707</span>  FROM % t1, % t2 </div> |
| <div class="line"><a name="l03708"></a><span class="lineno"> 3708</span>  WHERE t1.% = t2.id AND t1.% <> t2.class',</div> |
| <div class="line"><a name="l03709"></a><span class="lineno"> 3709</span>  ARRAY[</div> |
| <div class="line"><a name="l03710"></a><span class="lineno"> 3710</span>  encoded_table_name,</div> |
| <div class="line"><a name="l03711"></a><span class="lineno"> 3711</span>  result_table_name_final,</div> |
| <div class="line"><a name="l03712"></a><span class="lineno"> 3712</span>  id_col_name,</div> |
| <div class="line"><a name="l03713"></a><span class="lineno"> 3713</span>  class_col_name</div> |
| <div class="line"><a name="l03714"></a><span class="lineno"> 3714</span>  ]</div> |
| <div class="line"><a name="l03715"></a><span class="lineno"> 3715</span>  ) </div> |
| <div class="line"><a name="l03716"></a><span class="lineno"> 3716</span>  INTO curstmt;</div> |
| <div class="line"><a name="l03717"></a><span class="lineno"> 3717</span>  </div> |
| <div class="line"><a name="l03718"></a><span class="lineno"> 3718</span>  EXECUTE curstmt INTO mis_of_row;</div> |
| <div class="line"><a name="l03719"></a><span class="lineno"> 3719</span>  </div> |
| <div class="line"><a name="l03720"></a><span class="lineno"> 3720</span>  EXECUTE 'DROP TABLE IF EXISTS ' || encoded_table_name || ';';</div> |
| <div class="line"><a name="l03721"></a><span class="lineno"> 3721</span>  EXECUTE 'DROP TABLE IF EXISTS ' || result_table_name || ';';</div> |
| <div class="line"><a name="l03722"></a><span class="lineno"> 3722</span>  EXECUTE 'DROP TABLE IF EXISTS ' || result_table_name_final || ';'; </div> |
| <div class="line"><a name="l03723"></a><span class="lineno"> 3723</span>  RETURN (num_of_row - mis_of_row) / num_of_row;</div> |
| <div class="line"><a name="l03724"></a><span class="lineno"> 3724</span> END;</div> |
| <div class="line"><a name="l03725"></a><span class="lineno"> 3725</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l03726"></a><span class="lineno"> 3726</span> </div> |
| <div class="line"><a name="l03727"></a><span class="lineno"> 3727</span> </div> |
| <div class="line"><a name="l03728"></a><span class="lineno"> 3728</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l03729"></a><span class="lineno"> 3729</span> <span class="comment"> * @brief Cleanup the trained model table and any relevant tables.</span></div> |
| <div class="line"><a name="l03730"></a><span class="lineno"> 3730</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03731"></a><span class="lineno"> 3731</span> <span class="comment"> * @param model_table_name The name of the table containing</span></div> |
| <div class="line"><a name="l03732"></a><span class="lineno"> 3732</span> <span class="comment"> * the model's information.</span></div> |
| <div class="line"><a name="l03733"></a><span class="lineno"> 3733</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03734"></a><span class="lineno"> 3734</span> <span class="comment"> * @return The status of that cleanup operation.</span></div> |
| <div class="line"><a name="l03735"></a><span class="lineno"> 3735</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03736"></a><span class="lineno"> 3736</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l03737"></a><span class="lineno"> 3737</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__treemodel_clean</div> |
| <div class="line"><a name="l03738"></a><span class="lineno"> 3738</span>  ( </div> |
| <div class="line"><a name="l03739"></a><span class="lineno"> 3739</span>  model_table_name TEXT</div> |
| <div class="line"><a name="l03740"></a><span class="lineno"> 3740</span>  ) </div> |
| <div class="line"><a name="l03741"></a><span class="lineno"> 3741</span> RETURNS BOOLEAN AS $$</div> |
| <div class="line"><a name="l03742"></a><span class="lineno"> 3742</span> DECLARE</div> |
| <div class="line"><a name="l03743"></a><span class="lineno"> 3743</span>  metatable_name TEXT;</div> |
| <div class="line"><a name="l03744"></a><span class="lineno"> 3744</span>  ref_count INT;</div> |
| <div class="line"><a name="l03745"></a><span class="lineno"> 3745</span> BEGIN</div> |
| <div class="line"><a name="l03746"></a><span class="lineno"> 3746</span>  -- get rid of the messages whose severity level is lower than 'WARNING'</div> |
| <div class="line"><a name="l03747"></a><span class="lineno"> 3747</span>  SET client_min_messages = WARNING;</div> |
| <div class="line"><a name="l03748"></a><span class="lineno"> 3748</span>  </div> |
| <div class="line"><a name="l03749"></a><span class="lineno"> 3749</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03750"></a><span class="lineno"> 3750</span>  (</div> |
| <div class="line"><a name="l03751"></a><span class="lineno"> 3751</span>  (model_table_name IS NOT NULL) AND</div> |
| <div class="line"><a name="l03752"></a><span class="lineno"> 3752</span>  (</div> |
| <div class="line"><a name="l03753"></a><span class="lineno"> 3753</span>  MADLIB_SCHEMA.__table_exists</div> |
| <div class="line"><a name="l03754"></a><span class="lineno"> 3754</span>  (</div> |
| <div class="line"><a name="l03755"></a><span class="lineno"> 3755</span>  model_table_name</div> |
| <div class="line"><a name="l03756"></a><span class="lineno"> 3756</span>  )</div> |
| <div class="line"><a name="l03757"></a><span class="lineno"> 3757</span>  ),</div> |
| <div class="line"><a name="l03758"></a><span class="lineno"> 3758</span>  'the specified tree table' || </div> |
| <div class="line"><a name="l03759"></a><span class="lineno"> 3759</span>  coalesce('<' || </div> |
| <div class="line"><a name="l03760"></a><span class="lineno"> 3760</span>  model_table_name || </div> |
| <div class="line"><a name="l03761"></a><span class="lineno"> 3761</span>  '> does not exists', ' is NULL')</div> |
| <div class="line"><a name="l03762"></a><span class="lineno"> 3762</span>  ); </div> |
| <div class="line"><a name="l03763"></a><span class="lineno"> 3763</span>  </div> |
| <div class="line"><a name="l03764"></a><span class="lineno"> 3764</span>  IF (MADLIB_SCHEMA.__table_exists('MADLIB_SCHEMA.training_info')) THEN</div> |
| <div class="line"><a name="l03765"></a><span class="lineno"> 3765</span>  metatable_name = MADLIB_SCHEMA.__get_metatable_name(model_table_name);</div> |
| <div class="line"><a name="l03766"></a><span class="lineno"> 3766</span>  IF( metatable_name IS NOT NULL) THEN</div> |
| <div class="line"><a name="l03767"></a><span class="lineno"> 3767</span>  SELECT count(*) </div> |
| <div class="line"><a name="l03768"></a><span class="lineno"> 3768</span>  FROM MADLIB_SCHEMA.training_info</div> |
| <div class="line"><a name="l03769"></a><span class="lineno"> 3769</span>  WHERE training_metatable_oid = metatable_name::regclass</div> |
| <div class="line"><a name="l03770"></a><span class="lineno"> 3770</span>  INTO ref_count; </div> |
| <div class="line"><a name="l03771"></a><span class="lineno"> 3771</span>  </div> |
| <div class="line"><a name="l03772"></a><span class="lineno"> 3772</span>  -- if the metatable is not referenced by other training procedure.</div> |
| <div class="line"><a name="l03773"></a><span class="lineno"> 3773</span>  IF (ref_count = 1) THEN</div> |
| <div class="line"><a name="l03774"></a><span class="lineno"> 3774</span>  PERFORM MADLIB_SCHEMA.__drop_metatable(metatable_name);</div> |
| <div class="line"><a name="l03775"></a><span class="lineno"> 3775</span>  EXECUTE 'DROP TABLE IF EXISTS ' || </div> |
| <div class="line"><a name="l03776"></a><span class="lineno"> 3776</span>  MADLIB_SCHEMA.__get_encode_table_name(model_table_name) || ';';</div> |
| <div class="line"><a name="l03777"></a><span class="lineno"> 3777</span>  END IF;</div> |
| <div class="line"><a name="l03778"></a><span class="lineno"> 3778</span>  END IF;</div> |
| <div class="line"><a name="l03779"></a><span class="lineno"> 3779</span>  </div> |
| <div class="line"><a name="l03780"></a><span class="lineno"> 3780</span>  -- remove the record first, and then drop the table</div> |
| <div class="line"><a name="l03781"></a><span class="lineno"> 3781</span>  PERFORM MADLIB_SCHEMA.__delete_traininginfo(model_table_name);</div> |
| <div class="line"><a name="l03782"></a><span class="lineno"> 3782</span>  EXECUTE 'DROP TABLE IF EXISTS ' || model_table_name;</div> |
| <div class="line"><a name="l03783"></a><span class="lineno"> 3783</span>  </div> |
| <div class="line"><a name="l03784"></a><span class="lineno"> 3784</span>  ELSE</div> |
| <div class="line"><a name="l03785"></a><span class="lineno"> 3785</span>  EXECUTE 'DROP TABLE IF EXISTS ' || model_table_name;</div> |
| <div class="line"><a name="l03786"></a><span class="lineno"> 3786</span>  END IF;</div> |
| <div class="line"><a name="l03787"></a><span class="lineno"> 3787</span>  </div> |
| <div class="line"><a name="l03788"></a><span class="lineno"> 3788</span>  RETURN 't'; </div> |
| <div class="line"><a name="l03789"></a><span class="lineno"> 3789</span> END</div> |
| <div class="line"><a name="l03790"></a><span class="lineno"> 3790</span> $$ LANGUAGE PLPGSQL;</div> |
| <div class="line"><a name="l03791"></a><span class="lineno"> 3791</span> </div> |
| <div class="line"><a name="l03792"></a><span class="lineno"> 3792</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l03793"></a><span class="lineno"> 3793</span> <span class="comment"> * @brief Validate the common parameters for C4.5 and RF API.</span></div> |
| <div class="line"><a name="l03794"></a><span class="lineno"> 3794</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03795"></a><span class="lineno"> 3795</span> <span class="comment"> * @param split_criterion The name of the split criterion that should be used </span></div> |
| <div class="line"><a name="l03796"></a><span class="lineno"> 3796</span> <span class="comment"> * for tree construction. The valid values are</span></div> |
| <div class="line"><a name="l03797"></a><span class="lineno"> 3797</span> <span class="comment"> * ‘infogain’, ‘gainratio’, and ‘gini’. It can't be NULL.</span></div> |
| <div class="line"><a name="l03798"></a><span class="lineno"> 3798</span> <span class="comment"> * @param training_table_name The name of the table/view with the source data.</span></div> |
| <div class="line"><a name="l03799"></a><span class="lineno"> 3799</span> <span class="comment"> * @param result_table_name The name of the table where the resulting DT </span></div> |
| <div class="line"><a name="l03800"></a><span class="lineno"> 3800</span> <span class="comment"> * will be kept.</span></div> |
| <div class="line"><a name="l03801"></a><span class="lineno"> 3801</span> <span class="comment"> * @param continuous_feature_names A comma-separated list of the names of features whose values </span></div> |
| <div class="line"><a name="l03802"></a><span class="lineno"> 3802</span> <span class="comment"> * are continuous. The default is null, which means there are </span></div> |
| <div class="line"><a name="l03803"></a><span class="lineno"> 3803</span> <span class="comment"> * no continuous features in the training table.</span></div> |
| <div class="line"><a name="l03804"></a><span class="lineno"> 3804</span> <span class="comment"> * @param feature_col_names A comma-separated list of the names of table columns, each of</span></div> |
| <div class="line"><a name="l03805"></a><span class="lineno"> 3805</span> <span class="comment"> * which defines a feature. The default value is null, which means </span></div> |
| <div class="line"><a name="l03806"></a><span class="lineno"> 3806</span> <span class="comment"> * all the columns in the training table, except columns named </span></div> |
| <div class="line"><a name="l03807"></a><span class="lineno"> 3807</span> <span class="comment"> * ‘id’ and ‘class’, will be used as features.</span></div> |
| <div class="line"><a name="l03808"></a><span class="lineno"> 3808</span> <span class="comment"> * @param id_col_name The name of the column containing an ID for each record.</span></div> |
| <div class="line"><a name="l03809"></a><span class="lineno"> 3809</span> <span class="comment"> * @param class_col_name The name of the column containing the labeled class. </span></div> |
| <div class="line"><a name="l03810"></a><span class="lineno"> 3810</span> <span class="comment"> * @param how2handle_missing_value The way to handle missing value. The valid value </span></div> |
| <div class="line"><a name="l03811"></a><span class="lineno"> 3811</span> <span class="comment"> * is 'explicit' or 'ignore'.</span></div> |
| <div class="line"><a name="l03812"></a><span class="lineno"> 3812</span> <span class="comment"> * @param max_tree_depth Specifies the maximum number of levels in the result DT </span></div> |
| <div class="line"><a name="l03813"></a><span class="lineno"> 3813</span> <span class="comment"> * to avoid overgrown DTs. </span></div> |
| <div class="line"><a name="l03814"></a><span class="lineno"> 3814</span> <span class="comment"> * @param node_prune_threshold The minimum percentage of the number of records required in a</span></div> |
| <div class="line"><a name="l03815"></a><span class="lineno"> 3815</span> <span class="comment"> * child node. It can't be NULL. The range of it is in [0.0, 1.0].</span></div> |
| <div class="line"><a name="l03816"></a><span class="lineno"> 3816</span> <span class="comment"> * This threshold only applies to the non-root nodes. Therefore,</span></div> |
| <div class="line"><a name="l03817"></a><span class="lineno"> 3817</span> <span class="comment"> * if its value is 1, then the trained tree only has one node (the root node);</span></div> |
| <div class="line"><a name="l03818"></a><span class="lineno"> 3818</span> <span class="comment"> * if its value is 0, then no nodes will be pruned by this parameter.</span></div> |
| <div class="line"><a name="l03819"></a><span class="lineno"> 3819</span> <span class="comment"> * @param node_split_threshold The minimum percentage of the number of records required in a</span></div> |
| <div class="line"><a name="l03820"></a><span class="lineno"> 3820</span> <span class="comment"> * node in order for a further split to be possible.</span></div> |
| <div class="line"><a name="l03821"></a><span class="lineno"> 3821</span> <span class="comment"> * It can't be NULL. The range of it is in [0.0, 1.0].</span></div> |
| <div class="line"><a name="l03822"></a><span class="lineno"> 3822</span> <span class="comment"> * If it's value is 1, then the trained tree only has two levels, since</span></div> |
| <div class="line"><a name="l03823"></a><span class="lineno"> 3823</span> <span class="comment"> * only the root node can grow; if its value is 0, then trees can grow</span></div> |
| <div class="line"><a name="l03824"></a><span class="lineno"> 3824</span> <span class="comment"> * extensively.</span></div> |
| <div class="line"><a name="l03825"></a><span class="lineno"> 3825</span> <span class="comment"> * @param verbosity > 0 means this function runs in verbose mode. </span></div> |
| <div class="line"><a name="l03826"></a><span class="lineno"> 3826</span> <span class="comment"> * @param error_msg The reported error message when result_table_name is invalid.</span></div> |
| <div class="line"><a name="l03827"></a><span class="lineno"> 3827</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l03828"></a><span class="lineno"> 3828</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l03829"></a><span class="lineno"> 3829</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__check_dt_common_params</div> |
| <div class="line"><a name="l03830"></a><span class="lineno"> 3830</span>  (</div> |
| <div class="line"><a name="l03831"></a><span class="lineno"> 3831</span>  split_criterion TEXT,</div> |
| <div class="line"><a name="l03832"></a><span class="lineno"> 3832</span>  training_table_name TEXT, </div> |
| <div class="line"><a name="l03833"></a><span class="lineno"> 3833</span>  result_table_name TEXT,</div> |
| <div class="line"><a name="l03834"></a><span class="lineno"> 3834</span>  continuous_feature_names TEXT, </div> |
| <div class="line"><a name="l03835"></a><span class="lineno"> 3835</span>  feature_col_names TEXT, </div> |
| <div class="line"><a name="l03836"></a><span class="lineno"> 3836</span>  id_col_name TEXT, </div> |
| <div class="line"><a name="l03837"></a><span class="lineno"> 3837</span>  class_col_name TEXT, </div> |
| <div class="line"><a name="l03838"></a><span class="lineno"> 3838</span>  how2handle_missing_value TEXT,</div> |
| <div class="line"><a name="l03839"></a><span class="lineno"> 3839</span>  max_tree_depth INT,</div> |
| <div class="line"><a name="l03840"></a><span class="lineno"> 3840</span>  node_prune_threshold FLOAT,</div> |
| <div class="line"><a name="l03841"></a><span class="lineno"> 3841</span>  node_split_threshold FLOAT, </div> |
| <div class="line"><a name="l03842"></a><span class="lineno"> 3842</span>  verbosity INT,</div> |
| <div class="line"><a name="l03843"></a><span class="lineno"> 3843</span>  error_msg TEXT</div> |
| <div class="line"><a name="l03844"></a><span class="lineno"> 3844</span>  ) </div> |
| <div class="line"><a name="l03845"></a><span class="lineno"> 3845</span> RETURNS void AS $$</div> |
| <div class="line"><a name="l03846"></a><span class="lineno"> 3846</span> DECLARE</div> |
| <div class="line"><a name="l03847"></a><span class="lineno"> 3847</span>  num_of_element BIGINT;</div> |
| <div class="line"><a name="l03848"></a><span class="lineno"> 3848</span> BEGIN</div> |
| <div class="line"><a name="l03849"></a><span class="lineno"> 3849</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03850"></a><span class="lineno"> 3850</span>  (</div> |
| <div class="line"><a name="l03851"></a><span class="lineno"> 3851</span>  (split_criterion IS NOT NULL) AND</div> |
| <div class="line"><a name="l03852"></a><span class="lineno"> 3852</span>  (</div> |
| <div class="line"><a name="l03853"></a><span class="lineno"> 3853</span>  split_criterion = 'infogain' OR </div> |
| <div class="line"><a name="l03854"></a><span class="lineno"> 3854</span>  split_criterion = 'gainratio' OR </div> |
| <div class="line"><a name="l03855"></a><span class="lineno"> 3855</span>  split_criterion = 'gini'</div> |
| <div class="line"><a name="l03856"></a><span class="lineno"> 3856</span>  ),</div> |
| <div class="line"><a name="l03857"></a><span class="lineno"> 3857</span>  'split_criterion must be infogain, gainratio or gini'</div> |
| <div class="line"><a name="l03858"></a><span class="lineno"> 3858</span>  );</div> |
| <div class="line"><a name="l03859"></a><span class="lineno"> 3859</span> </div> |
| <div class="line"><a name="l03860"></a><span class="lineno"> 3860</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03861"></a><span class="lineno"> 3861</span>  (</div> |
| <div class="line"><a name="l03862"></a><span class="lineno"> 3862</span>  how2handle_missing_value = 'ignore' OR </div> |
| <div class="line"><a name="l03863"></a><span class="lineno"> 3863</span>  how2handle_missing_value = 'explicit',</div> |
| <div class="line"><a name="l03864"></a><span class="lineno"> 3864</span>  'how2handle_missing_value must be ignore or explicit!'</div> |
| <div class="line"><a name="l03865"></a><span class="lineno"> 3865</span>  ); </div> |
| <div class="line"><a name="l03866"></a><span class="lineno"> 3866</span> </div> |
| <div class="line"><a name="l03867"></a><span class="lineno"> 3867</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03868"></a><span class="lineno"> 3868</span>  (</div> |
| <div class="line"><a name="l03869"></a><span class="lineno"> 3869</span>  max_tree_depth IS NOT NULL AND</div> |
| <div class="line"><a name="l03870"></a><span class="lineno"> 3870</span>  max_tree_depth > 0,</div> |
| <div class="line"><a name="l03871"></a><span class="lineno"> 3871</span>  'max_tree_depth value must be greater than 0'</div> |
| <div class="line"><a name="l03872"></a><span class="lineno"> 3872</span>  );</div> |
| <div class="line"><a name="l03873"></a><span class="lineno"> 3873</span> </div> |
| <div class="line"><a name="l03874"></a><span class="lineno"> 3874</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03875"></a><span class="lineno"> 3875</span>  (</div> |
| <div class="line"><a name="l03876"></a><span class="lineno"> 3876</span>  node_prune_threshold IS NOT NULL AND</div> |
| <div class="line"><a name="l03877"></a><span class="lineno"> 3877</span>  float8ge(node_prune_threshold, 0) AND</div> |
| <div class="line"><a name="l03878"></a><span class="lineno"> 3878</span>  float8le(node_prune_threshold, 1),</div> |
| <div class="line"><a name="l03879"></a><span class="lineno"> 3879</span>  'node_prune_threshold value must be in range from 0 to 1'</div> |
| <div class="line"><a name="l03880"></a><span class="lineno"> 3880</span>  );</div> |
| <div class="line"><a name="l03881"></a><span class="lineno"> 3881</span> </div> |
| <div class="line"><a name="l03882"></a><span class="lineno"> 3882</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03883"></a><span class="lineno"> 3883</span>  (</div> |
| <div class="line"><a name="l03884"></a><span class="lineno"> 3884</span>  node_split_threshold IS NOT NULL AND </div> |
| <div class="line"><a name="l03885"></a><span class="lineno"> 3885</span>  float8ge(node_split_threshold, 0) AND</div> |
| <div class="line"><a name="l03886"></a><span class="lineno"> 3886</span>  float8le(node_split_threshold, 1), </div> |
| <div class="line"><a name="l03887"></a><span class="lineno"> 3887</span>  'node_split_threshold value must be in range from 0 to 1'</div> |
| <div class="line"><a name="l03888"></a><span class="lineno"> 3888</span>  );</div> |
| <div class="line"><a name="l03889"></a><span class="lineno"> 3889</span>  </div> |
| <div class="line"><a name="l03890"></a><span class="lineno"> 3890</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03891"></a><span class="lineno"> 3891</span>  (</div> |
| <div class="line"><a name="l03892"></a><span class="lineno"> 3892</span>  verbosity IS NOT NULL, </div> |
| <div class="line"><a name="l03893"></a><span class="lineno"> 3893</span>  'verbosity must be non-null'</div> |
| <div class="line"><a name="l03894"></a><span class="lineno"> 3894</span>  );</div> |
| <div class="line"><a name="l03895"></a><span class="lineno"> 3895</span>  </div> |
| <div class="line"><a name="l03896"></a><span class="lineno"> 3896</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03897"></a><span class="lineno"> 3897</span>  (</div> |
| <div class="line"><a name="l03898"></a><span class="lineno"> 3898</span>  id_col_name IS NOT NULL AND</div> |
| <div class="line"><a name="l03899"></a><span class="lineno"> 3899</span>  class_col_name IS NOT NULL AND</div> |
| <div class="line"><a name="l03900"></a><span class="lineno"> 3900</span>  length(btrim(id_col_name, ' ')) > 0 AND</div> |
| <div class="line"><a name="l03901"></a><span class="lineno"> 3901</span>  length(btrim(class_col_name, ' ')) > 0,</div> |
| <div class="line"><a name="l03902"></a><span class="lineno"> 3902</span>  'invalid id column name or class column name'</div> |
| <div class="line"><a name="l03903"></a><span class="lineno"> 3903</span>  );</div> |
| <div class="line"><a name="l03904"></a><span class="lineno"> 3904</span> </div> |
| <div class="line"><a name="l03905"></a><span class="lineno"> 3905</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03906"></a><span class="lineno"> 3906</span>  (</div> |
| <div class="line"><a name="l03907"></a><span class="lineno"> 3907</span>  training_table_name IS NOT NULL AND</div> |
| <div class="line"><a name="l03908"></a><span class="lineno"> 3908</span>  MADLIB_SCHEMA.__table_exists</div> |
| <div class="line"><a name="l03909"></a><span class="lineno"> 3909</span>  (</div> |
| <div class="line"><a name="l03910"></a><span class="lineno"> 3910</span>  training_table_name</div> |
| <div class="line"><a name="l03911"></a><span class="lineno"> 3911</span>  ),</div> |
| <div class="line"><a name="l03912"></a><span class="lineno"> 3912</span>  'the specified training table' || </div> |
| <div class="line"><a name="l03913"></a><span class="lineno"> 3913</span>  coalesce('<' || </div> |
| <div class="line"><a name="l03914"></a><span class="lineno"> 3914</span>  training_table_name || </div> |
| <div class="line"><a name="l03915"></a><span class="lineno"> 3915</span>  '> does not exist', ' is NULL')</div> |
| <div class="line"><a name="l03916"></a><span class="lineno"> 3916</span>  );</div> |
| <div class="line"><a name="l03917"></a><span class="lineno"> 3917</span> </div> |
| <div class="line"><a name="l03918"></a><span class="lineno"> 3918</span>  EXECUTE 'SELECT count(*) FROM</div> |
| <div class="line"><a name="l03919"></a><span class="lineno"> 3919</span>  (SELECT * FROM '||training_table_name||' LIMIT 1) l' </div> |
| <div class="line"><a name="l03920"></a><span class="lineno"> 3920</span>  INTO num_of_element;</div> |
| <div class="line"><a name="l03921"></a><span class="lineno"> 3921</span> </div> |
| <div class="line"><a name="l03922"></a><span class="lineno"> 3922</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03923"></a><span class="lineno"> 3923</span>  (</div> |
| <div class="line"><a name="l03924"></a><span class="lineno"> 3924</span>  num_of_element > 0,</div> |
| <div class="line"><a name="l03925"></a><span class="lineno"> 3925</span>  'the specified training table <'||training_table_name||</div> |
| <div class="line"><a name="l03926"></a><span class="lineno"> 3926</span>  '> should not be empty'</div> |
| <div class="line"><a name="l03927"></a><span class="lineno"> 3927</span>  );</div> |
| <div class="line"><a name="l03928"></a><span class="lineno"> 3928</span> </div> |
| <div class="line"><a name="l03929"></a><span class="lineno"> 3929</span>  </div> |
| <div class="line"><a name="l03930"></a><span class="lineno"> 3930</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03931"></a><span class="lineno"> 3931</span>  (</div> |
| <div class="line"><a name="l03932"></a><span class="lineno"> 3932</span>  result_table_name IS NOT NULL,</div> |
| <div class="line"><a name="l03933"></a><span class="lineno"> 3933</span>  'the specified result ' || error_msg || ' table name is NULL'</div> |
| <div class="line"><a name="l03934"></a><span class="lineno"> 3934</span>  );</div> |
| <div class="line"><a name="l03935"></a><span class="lineno"> 3935</span>  </div> |
| <div class="line"><a name="l03936"></a><span class="lineno"> 3936</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03937"></a><span class="lineno"> 3937</span>  (</div> |
| <div class="line"><a name="l03938"></a><span class="lineno"> 3938</span>  NOT MADLIB_SCHEMA.__table_exists</div> |
| <div class="line"><a name="l03939"></a><span class="lineno"> 3939</span>  (</div> |
| <div class="line"><a name="l03940"></a><span class="lineno"> 3940</span>  result_table_name </div> |
| <div class="line"><a name="l03941"></a><span class="lineno"> 3941</span>  )</div> |
| <div class="line"><a name="l03942"></a><span class="lineno"> 3942</span>  ,</div> |
| <div class="line"><a name="l03943"></a><span class="lineno"> 3943</span>  'the specified result ' || error_msg || ' table<' || </div> |
| <div class="line"><a name="l03944"></a><span class="lineno"> 3944</span>  result_table_name || </div> |
| <div class="line"><a name="l03945"></a><span class="lineno"> 3945</span>  '> exists' </div> |
| <div class="line"><a name="l03946"></a><span class="lineno"> 3946</span>  );</div> |
| <div class="line"><a name="l03947"></a><span class="lineno"> 3947</span> END</div> |
| <div class="line"><a name="l03948"></a><span class="lineno"> 3948</span> $$ LANGUAGE PLPGSQL STABLE;</div> |
| <div class="line"><a name="l03949"></a><span class="lineno"> 3949</span> </div> |
| <div class="line"><a name="l03950"></a><span class="lineno"> 3950</span> </div> |
| <div class="line"><a name="l03951"></a><span class="lineno"> 3951</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l03952"></a><span class="lineno"> 3952</span> <span class="comment"> * @brief Get the name of the encoded table and the name of</span></div> |
| <div class="line"><a name="l03953"></a><span class="lineno"> 3953</span> <span class="comment"> * its meta table.</span></div> |
| <div class="line"><a name="l03954"></a><span class="lineno"> 3954</span> <span class="comment"> * @param result_table_name The name of the table where the </span></div> |
| <div class="line"><a name="l03955"></a><span class="lineno"> 3955</span> <span class="comment"> * resulting DT will be kept </span></div> |
| <div class="line"><a name="l03956"></a><span class="lineno"> 3956</span> <span class="comment"> * @param error_msg The reported error message when the</span></div> |
| <div class="line"><a name="l03957"></a><span class="lineno"> 3957</span> <span class="comment"> * length of result schema name plus</span></div> |
| <div class="line"><a name="l03958"></a><span class="lineno"> 3958</span> <span class="comment"> * the length of result table name is</span></div> |
| <div class="line"><a name="l03959"></a><span class="lineno"> 3959</span> <span class="comment"> * larger than 58.</span></div> |
| <div class="line"><a name="l03960"></a><span class="lineno"> 3960</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l03961"></a><span class="lineno"> 3961</span> <span class="comment"> * @return A text array that contains two elements. The firest element</span></div> |
| <div class="line"><a name="l03962"></a><span class="lineno"> 3962</span> <span class="comment"> * is the encoded table name and the second is the meta table name.</span></div> |
| <div class="line"><a name="l03963"></a><span class="lineno"> 3963</span> <span class="comment"> * </span></div> |
| <div class="line"><a name="l03964"></a><span class="lineno"> 3964</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l03965"></a><span class="lineno"> 3965</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__gen_enc_meta_names</div> |
| <div class="line"><a name="l03966"></a><span class="lineno"> 3966</span>  (</div> |
| <div class="line"><a name="l03967"></a><span class="lineno"> 3967</span>  result_table_name TEXT,</div> |
| <div class="line"><a name="l03968"></a><span class="lineno"> 3968</span>  error_msg TEXT</div> |
| <div class="line"><a name="l03969"></a><span class="lineno"> 3969</span>  ) </div> |
| <div class="line"><a name="l03970"></a><span class="lineno"> 3970</span> RETURNS TEXT[] AS $$</div> |
| <div class="line"><a name="l03971"></a><span class="lineno"> 3971</span> DECLARE</div> |
| <div class="line"><a name="l03972"></a><span class="lineno"> 3972</span>  result_schema_name TEXT;</div> |
| <div class="line"><a name="l03973"></a><span class="lineno"> 3973</span>  table_names TEXT[];</div> |
| <div class="line"><a name="l03974"></a><span class="lineno"> 3974</span> BEGIN</div> |
| <div class="line"><a name="l03975"></a><span class="lineno"> 3975</span>  result_schema_name = MADLIB_SCHEMA.__get_schema_name(result_table_name);</div> |
| <div class="line"><a name="l03976"></a><span class="lineno"> 3976</span>  </div> |
| <div class="line"><a name="l03977"></a><span class="lineno"> 3977</span>  -- the maximum length of an identifier 63</div> |
| <div class="line"><a name="l03978"></a><span class="lineno"> 3978</span>  -- encoding table name convension: <schema name>_<table name>_ed</div> |
| <div class="line"><a name="l03979"></a><span class="lineno"> 3979</span>  -- data info table name convension: <schema name>_<table name>_di</div> |
| <div class="line"><a name="l03980"></a><span class="lineno"> 3980</span>  -- the KV table name convension: <schema name>_<table name>_<####></div> |
| <div class="line"><a name="l03981"></a><span class="lineno"> 3981</span>  -- therefore, the maximum length of '<schema name>_<table name>' is 58</div> |
| <div class="line"><a name="l03982"></a><span class="lineno"> 3982</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l03983"></a><span class="lineno"> 3983</span>  (</div> |
| <div class="line"><a name="l03984"></a><span class="lineno"> 3984</span>  length(</div> |
| <div class="line"><a name="l03985"></a><span class="lineno"> 3985</span>  result_schema_name || </div> |
| <div class="line"><a name="l03986"></a><span class="lineno"> 3986</span>  '_' || </div> |
| <div class="line"><a name="l03987"></a><span class="lineno"> 3987</span>  result_table_name) <= 58,</div> |
| <div class="line"><a name="l03988"></a><span class="lineno"> 3988</span>  'the maximum length of ''' || error_msg || ''' is 58'</div> |
| <div class="line"><a name="l03989"></a><span class="lineno"> 3989</span>  );</div> |
| <div class="line"><a name="l03990"></a><span class="lineno"> 3990</span> </div> |
| <div class="line"><a name="l03991"></a><span class="lineno"> 3991</span>  -- the encoded table and meta table will be under the specified schema</div> |
| <div class="line"><a name="l03992"></a><span class="lineno"> 3992</span>  table_names[1] = result_schema_name || </div> |
| <div class="line"><a name="l03993"></a><span class="lineno"> 3993</span>  '.' || </div> |
| <div class="line"><a name="l03994"></a><span class="lineno"> 3994</span>  replace(result_table_name, '.', '_') || </div> |
| <div class="line"><a name="l03995"></a><span class="lineno"> 3995</span>  '_ed';</div> |
| <div class="line"><a name="l03996"></a><span class="lineno"> 3996</span>  table_names[2] = result_schema_name || </div> |
| <div class="line"><a name="l03997"></a><span class="lineno"> 3997</span>  '.' || </div> |
| <div class="line"><a name="l03998"></a><span class="lineno"> 3998</span>  replace(result_table_name, '.', '_') || </div> |
| <div class="line"><a name="l03999"></a><span class="lineno"> 3999</span>  '_di'; </div> |
| <div class="line"><a name="l04000"></a><span class="lineno"> 4000</span>  RETURN table_names;</div> |
| <div class="line"><a name="l04001"></a><span class="lineno"> 4001</span> END</div> |
| <div class="line"><a name="l04002"></a><span class="lineno"> 4002</span> $$ LANGUAGE PLPGSQL STABLE;</div> |
| <div class="line"><a name="l04003"></a><span class="lineno"> 4003</span> </div> |
| <div class="line"><a name="l04004"></a><span class="lineno"> 4004</span> </div> |
| <div class="line"><a name="l04005"></a><span class="lineno"> 4005</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l04006"></a><span class="lineno"> 4006</span> <span class="comment"> * @brief Validate if the provided columns are in the training table or not.</span></div> |
| <div class="line"><a name="l04007"></a><span class="lineno"> 4007</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l04008"></a><span class="lineno"> 4008</span> <span class="comment"> * @param training_table_name The name of the table/view with the source data.</span></div> |
| <div class="line"><a name="l04009"></a><span class="lineno"> 4009</span> <span class="comment"> * @param continuous_feature_names A text array that contains all the continuous </span></div> |
| <div class="line"><a name="l04010"></a><span class="lineno"> 4010</span> <span class="comment"> * features' names. </span></div> |
| <div class="line"><a name="l04011"></a><span class="lineno"> 4011</span> <span class="comment"> * @param feature_col_names A text array that contains all the features' names.</span></div> |
| <div class="line"><a name="l04012"></a><span class="lineno"> 4012</span> <span class="comment"> * @param id_col_name The name of the column containing an ID for each record.</span></div> |
| <div class="line"><a name="l04013"></a><span class="lineno"> 4013</span> <span class="comment"> * @param class_col_name The name of the column containing the labeled class. </span></div> |
| <div class="line"><a name="l04014"></a><span class="lineno"> 4014</span> <span class="comment"> * @param features_per_node The number of features to be considered when finding </span></div> |
| <div class="line"><a name="l04015"></a><span class="lineno"> 4015</span> <span class="comment"> * a best split.</span></div> |
| <div class="line"><a name="l04016"></a><span class="lineno"> 4016</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l04017"></a><span class="lineno"> 4017</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__check_training_table</div> |
| <div class="line"><a name="l04018"></a><span class="lineno"> 4018</span>  (</div> |
| <div class="line"><a name="l04019"></a><span class="lineno"> 4019</span>  training_table_name TEXT,</div> |
| <div class="line"><a name="l04020"></a><span class="lineno"> 4020</span>  continuous_feature_names TEXT[], </div> |
| <div class="line"><a name="l04021"></a><span class="lineno"> 4021</span>  feature_col_names TEXT[], </div> |
| <div class="line"><a name="l04022"></a><span class="lineno"> 4022</span>  id_col_name TEXT, </div> |
| <div class="line"><a name="l04023"></a><span class="lineno"> 4023</span>  class_col_name TEXT,</div> |
| <div class="line"><a name="l04024"></a><span class="lineno"> 4024</span>  features_per_node INT</div> |
| <div class="line"><a name="l04025"></a><span class="lineno"> 4025</span>  ) </div> |
| <div class="line"><a name="l04026"></a><span class="lineno"> 4026</span> RETURNS VOID AS $$</div> |
| <div class="line"><a name="l04027"></a><span class="lineno"> 4027</span> DECLARE</div> |
| <div class="line"><a name="l04028"></a><span class="lineno"> 4028</span>  num_attrs INT;</div> |
| <div class="line"><a name="l04029"></a><span class="lineno"> 4029</span> BEGIN</div> |
| <div class="line"><a name="l04030"></a><span class="lineno"> 4030</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l04031"></a><span class="lineno"> 4031</span>  (</div> |
| <div class="line"><a name="l04032"></a><span class="lineno"> 4032</span>  MADLIB_SCHEMA.__column_exists</div> |
| <div class="line"><a name="l04033"></a><span class="lineno"> 4033</span>  (</div> |
| <div class="line"><a name="l04034"></a><span class="lineno"> 4034</span>  training_table_name,</div> |
| <div class="line"><a name="l04035"></a><span class="lineno"> 4035</span>  lower(btrim(id_col_name, ' '))</div> |
| <div class="line"><a name="l04036"></a><span class="lineno"> 4036</span>  ),</div> |
| <div class="line"><a name="l04037"></a><span class="lineno"> 4037</span>  'the specified training table<' || </div> |
| <div class="line"><a name="l04038"></a><span class="lineno"> 4038</span>  training_table_name || </div> |
| <div class="line"><a name="l04039"></a><span class="lineno"> 4039</span>  '> does not have column ''' || </div> |
| <div class="line"><a name="l04040"></a><span class="lineno"> 4040</span>  id_col_name || </div> |
| <div class="line"><a name="l04041"></a><span class="lineno"> 4041</span>  ''''</div> |
| <div class="line"><a name="l04042"></a><span class="lineno"> 4042</span>  );</div> |
| <div class="line"><a name="l04043"></a><span class="lineno"> 4043</span> </div> |
| <div class="line"><a name="l04044"></a><span class="lineno"> 4044</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l04045"></a><span class="lineno"> 4045</span>  (</div> |
| <div class="line"><a name="l04046"></a><span class="lineno"> 4046</span>  MADLIB_SCHEMA.__column_exists</div> |
| <div class="line"><a name="l04047"></a><span class="lineno"> 4047</span>  (</div> |
| <div class="line"><a name="l04048"></a><span class="lineno"> 4048</span>  training_table_name,</div> |
| <div class="line"><a name="l04049"></a><span class="lineno"> 4049</span>  lower(btrim(class_col_name, ' '))</div> |
| <div class="line"><a name="l04050"></a><span class="lineno"> 4050</span>  ),</div> |
| <div class="line"><a name="l04051"></a><span class="lineno"> 4051</span>  'the specified training table<' || </div> |
| <div class="line"><a name="l04052"></a><span class="lineno"> 4052</span>  training_table_name || </div> |
| <div class="line"><a name="l04053"></a><span class="lineno"> 4053</span>  '> does not have column ''' || </div> |
| <div class="line"><a name="l04054"></a><span class="lineno"> 4054</span>  class_col_name || </div> |
| <div class="line"><a name="l04055"></a><span class="lineno"> 4055</span>  ''''</div> |
| <div class="line"><a name="l04056"></a><span class="lineno"> 4056</span>  );</div> |
| <div class="line"><a name="l04057"></a><span class="lineno"> 4057</span> </div> |
| <div class="line"><a name="l04058"></a><span class="lineno"> 4058</span>  IF (feature_col_names IS NULL) THEN</div> |
| <div class="line"><a name="l04059"></a><span class="lineno"> 4059</span>  -- 2 means the id and class column</div> |
| <div class="line"><a name="l04060"></a><span class="lineno"> 4060</span>  num_attrs = MADLIB_SCHEMA.__num_of_columns(training_table_name) - 2;</div> |
| <div class="line"><a name="l04061"></a><span class="lineno"> 4061</span>  </div> |
| <div class="line"><a name="l04062"></a><span class="lineno"> 4062</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l04063"></a><span class="lineno"> 4063</span>  (</div> |
| <div class="line"><a name="l04064"></a><span class="lineno"> 4064</span>  (features_per_node IS NULL AND num_attrs > 0) OR</div> |
| <div class="line"><a name="l04065"></a><span class="lineno"> 4065</span>  (features_per_node IS NOT NULL AND num_attrs >= features_per_node),</div> |
| <div class="line"><a name="l04066"></a><span class="lineno"> 4066</span>  'the value of features_per_node must be less than or equal to the total number ' ||</div> |
| <div class="line"><a name="l04067"></a><span class="lineno"> 4067</span>  'of features of the training table'</div> |
| <div class="line"><a name="l04068"></a><span class="lineno"> 4068</span>  ); </div> |
| <div class="line"><a name="l04069"></a><span class="lineno"> 4069</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l04070"></a><span class="lineno"> 4070</span>  (</div> |
| <div class="line"><a name="l04071"></a><span class="lineno"> 4071</span>  MADLIB_SCHEMA.__columns_in_table(continuous_feature_names, training_table_name),</div> |
| <div class="line"><a name="l04072"></a><span class="lineno"> 4072</span>  'each feature in continuous_feature_names must be a column of the training table'</div> |
| <div class="line"><a name="l04073"></a><span class="lineno"> 4073</span>  );</div> |
| <div class="line"><a name="l04074"></a><span class="lineno"> 4074</span>  ELSE</div> |
| <div class="line"><a name="l04075"></a><span class="lineno"> 4075</span>  num_attrs = array_upper(feature_col_names, 1);</div> |
| <div class="line"><a name="l04076"></a><span class="lineno"> 4076</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l04077"></a><span class="lineno"> 4077</span>  (</div> |
| <div class="line"><a name="l04078"></a><span class="lineno"> 4078</span>  (features_per_node IS NULL AND num_attrs > 0) OR</div> |
| <div class="line"><a name="l04079"></a><span class="lineno"> 4079</span>  (features_per_node IS NOT NULL AND num_attrs >= features_per_node),</div> |
| <div class="line"><a name="l04080"></a><span class="lineno"> 4080</span>  'the value of features_per_node must be less than or equal to the total number ' ||</div> |
| <div class="line"><a name="l04081"></a><span class="lineno"> 4081</span>  'of features of the training table'</div> |
| <div class="line"><a name="l04082"></a><span class="lineno"> 4082</span>  ); </div> |
| <div class="line"><a name="l04083"></a><span class="lineno"> 4083</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l04084"></a><span class="lineno"> 4084</span>  (</div> |
| <div class="line"><a name="l04085"></a><span class="lineno"> 4085</span>  MADLIB_SCHEMA.__columns_in_table(feature_col_names, training_table_name),</div> |
| <div class="line"><a name="l04086"></a><span class="lineno"> 4086</span>  'each feature in feature_col_names must be a column of the training table'</div> |
| <div class="line"><a name="l04087"></a><span class="lineno"> 4087</span>  );</div> |
| <div class="line"><a name="l04088"></a><span class="lineno"> 4088</span> </div> |
| <div class="line"><a name="l04089"></a><span class="lineno"> 4089</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l04090"></a><span class="lineno"> 4090</span>  (</div> |
| <div class="line"><a name="l04091"></a><span class="lineno"> 4091</span>  coalesce(continuous_feature_names, '{}'::TEXT[]) <@ feature_col_names,</div> |
| <div class="line"><a name="l04092"></a><span class="lineno"> 4092</span>  'each feature in continuous_feature_names must be in the feature_col_names'</div> |
| <div class="line"><a name="l04093"></a><span class="lineno"> 4093</span>  );</div> |
| <div class="line"><a name="l04094"></a><span class="lineno"> 4094</span>  END IF;</div> |
| <div class="line"><a name="l04095"></a><span class="lineno"> 4095</span> END</div> |
| <div class="line"><a name="l04096"></a><span class="lineno"> 4096</span> $$ LANGUAGE PLPGSQL STABLE;</div> |
| <div class="line"><a name="l04097"></a><span class="lineno"> 4097</span> </div> |
| <div class="line"><a name="l04098"></a><span class="lineno"> 4098</span> </div> |
| <div class="line"><a name="l04099"></a><span class="lineno"> 4099</span> <span class="comment">/* @ brief If the training table is a valid encoded table, then we use it directly.</span></div> |
| <div class="line"><a name="l04100"></a><span class="lineno"> 4100</span> <span class="comment"> * If the training table is not encoded, then we invoke the encoding procedure</span></div> |
| <div class="line"><a name="l04101"></a><span class="lineno"> 4101</span> <span class="comment"> * to transform the training table. </span></div> |
| <div class="line"><a name="l04102"></a><span class="lineno"> 4102</span> <span class="comment"> * With the encoded table, we call the tree grow engine to generate the final tree.</span></div> |
| <div class="line"><a name="l04103"></a><span class="lineno"> 4103</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l04104"></a><span class="lineno"> 4104</span> <span class="comment"> * @param dt_algo_name The name of the algorithom. Currently, it's</span></div> |
| <div class="line"><a name="l04105"></a><span class="lineno"> 4105</span> <span class="comment"> * 'C4.5' or 'RF'</span></div> |
| <div class="line"><a name="l04106"></a><span class="lineno"> 4106</span> <span class="comment"> * @param split_criterion This parameter specifies which split criterion </span></div> |
| <div class="line"><a name="l04107"></a><span class="lineno"> 4107</span> <span class="comment"> * should be used for tree construction and </span></div> |
| <div class="line"><a name="l04108"></a><span class="lineno"> 4108</span> <span class="comment"> * pruning. The valid values are infogain, </span></div> |
| <div class="line"><a name="l04109"></a><span class="lineno"> 4109</span> <span class="comment"> * gainratio, and gini.</span></div> |
| <div class="line"><a name="l04110"></a><span class="lineno"> 4110</span> <span class="comment"> * @param num_trees Total number of trees to be trained. </span></div> |
| <div class="line"><a name="l04111"></a><span class="lineno"> 4111</span> <span class="comment"> * @param features_per_node Total number of features used to compute split </span></div> |
| <div class="line"><a name="l04112"></a><span class="lineno"> 4112</span> <span class="comment"> * gain for each node. </span></div> |
| <div class="line"><a name="l04113"></a><span class="lineno"> 4113</span> <span class="comment"> * @param training_table_name The name of the table/view with the source data. </span></div> |
| <div class="line"><a name="l04114"></a><span class="lineno"> 4114</span> <span class="comment"> * @param validation_table_name The name of the validation table. </span></div> |
| <div class="line"><a name="l04115"></a><span class="lineno"> 4115</span> <span class="comment"> * @param tree_table_name The name of the table where the resulting </span></div> |
| <div class="line"><a name="l04116"></a><span class="lineno"> 4116</span> <span class="comment"> * DT/RF will be stored. </span></div> |
| <div class="line"><a name="l04117"></a><span class="lineno"> 4117</span> <span class="comment"> * @param continuous_feature_names A comma-separated list of the names of features whose values </span></div> |
| <div class="line"><a name="l04118"></a><span class="lineno"> 4118</span> <span class="comment"> * are continuous. The default is null, which means there are </span></div> |
| <div class="line"><a name="l04119"></a><span class="lineno"> 4119</span> <span class="comment"> * no continuous features in the training table.</span></div> |
| <div class="line"><a name="l04120"></a><span class="lineno"> 4120</span> <span class="comment"> * @param feature_col_names A comma-separated list of the names of table columns, each of</span></div> |
| <div class="line"><a name="l04121"></a><span class="lineno"> 4121</span> <span class="comment"> * which defines a feature. The default value is null, which means </span></div> |
| <div class="line"><a name="l04122"></a><span class="lineno"> 4122</span> <span class="comment"> * all the columns in the training table, except columns named </span></div> |
| <div class="line"><a name="l04123"></a><span class="lineno"> 4123</span> <span class="comment"> * ‘id’ and ‘class’, will be used as features.</span></div> |
| <div class="line"><a name="l04124"></a><span class="lineno"> 4124</span> <span class="comment"> * @param id_col_name The name of the column containing id of each point. </span></div> |
| <div class="line"><a name="l04125"></a><span class="lineno"> 4125</span> <span class="comment"> * @param class_col_name The name of the column containing correct class </span></div> |
| <div class="line"><a name="l04126"></a><span class="lineno"> 4126</span> <span class="comment"> * of each point. </span></div> |
| <div class="line"><a name="l04127"></a><span class="lineno"> 4127</span> <span class="comment"> * @param confidence_level A statistical confidence interval of the </span></div> |
| <div class="line"><a name="l04128"></a><span class="lineno"> 4128</span> <span class="comment"> * resubstitution error. </span></div> |
| <div class="line"><a name="l04129"></a><span class="lineno"> 4129</span> <span class="comment"> * @param how2handle_missing_value The way to handle missing value. The valid value </span></div> |
| <div class="line"><a name="l04130"></a><span class="lineno"> 4130</span> <span class="comment"> * is 'explicit' or 'ignore'.</span></div> |
| <div class="line"><a name="l04131"></a><span class="lineno"> 4131</span> <span class="comment"> * @param max_tree_depth Maximum decision tree depth. </span></div> |
| <div class="line"><a name="l04132"></a><span class="lineno"> 4132</span> <span class="comment"> * @param sampling_percentage The percentage of records sampled to train a tree.</span></div> |
| <div class="line"><a name="l04133"></a><span class="lineno"> 4133</span> <span class="comment"> * If it's NULL, 0.632 bootstrap will be used</span></div> |
| <div class="line"><a name="l04134"></a><span class="lineno"> 4134</span> <span class="comment"> * @param sampling_needed Whether enabling the sampling functionality. </span></div> |
| <div class="line"><a name="l04135"></a><span class="lineno"> 4135</span> <span class="comment"> * @param node_prune_threshold Specifies the minimum number of samples required </span></div> |
| <div class="line"><a name="l04136"></a><span class="lineno"> 4136</span> <span class="comment"> * in a child node. </span></div> |
| <div class="line"><a name="l04137"></a><span class="lineno"> 4137</span> <span class="comment"> * @param node_split_threshold Specifies the minimum number of samples required </span></div> |
| <div class="line"><a name="l04138"></a><span class="lineno"> 4138</span> <span class="comment"> * in a node in order for a further split </span></div> |
| <div class="line"><a name="l04139"></a><span class="lineno"> 4139</span> <span class="comment"> * to be possible. </span></div> |
| <div class="line"><a name="l04140"></a><span class="lineno"> 4140</span> <span class="comment"> * @param error_msg The reported error message when the result table</span></div> |
| <div class="line"><a name="l04141"></a><span class="lineno"> 4141</span> <span class="comment"> * name is invalid.</span></div> |
| <div class="line"><a name="l04142"></a><span class="lineno"> 4142</span> <span class="comment"> * @param verbosity > 0 means this function runs in verbose mode. </span></div> |
| <div class="line"><a name="l04143"></a><span class="lineno"> 4143</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l04144"></a><span class="lineno"> 4144</span> <span class="comment"> * @return An instance of __train_result.</span></div> |
| <div class="line"><a name="l04145"></a><span class="lineno"> 4145</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l04146"></a><span class="lineno"> 4146</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l04147"></a><span class="lineno"> 4147</span> CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__encode_and_train</div> |
| <div class="line"><a name="l04148"></a><span class="lineno"> 4148</span>  (</div> |
| <div class="line"><a name="l04149"></a><span class="lineno"> 4149</span>  dt_algo_name TEXT,</div> |
| <div class="line"><a name="l04150"></a><span class="lineno"> 4150</span>  split_criterion TEXT,</div> |
| <div class="line"><a name="l04151"></a><span class="lineno"> 4151</span>  num_trees INT,</div> |
| <div class="line"><a name="l04152"></a><span class="lineno"> 4152</span>  features_per_node INT,</div> |
| <div class="line"><a name="l04153"></a><span class="lineno"> 4153</span>  training_table_name TEXT,</div> |
| <div class="line"><a name="l04154"></a><span class="lineno"> 4154</span>  validation_table_name TEXT,</div> |
| <div class="line"><a name="l04155"></a><span class="lineno"> 4155</span>  tree_table_name TEXT,</div> |
| <div class="line"><a name="l04156"></a><span class="lineno"> 4156</span>  continuous_feature_names TEXT,</div> |
| <div class="line"><a name="l04157"></a><span class="lineno"> 4157</span>  feature_col_names TEXT, </div> |
| <div class="line"><a name="l04158"></a><span class="lineno"> 4158</span>  id_col_name TEXT, </div> |
| <div class="line"><a name="l04159"></a><span class="lineno"> 4159</span>  class_col_name TEXT, </div> |
| <div class="line"><a name="l04160"></a><span class="lineno"> 4160</span>  confidence_level FLOAT8, </div> |
| <div class="line"><a name="l04161"></a><span class="lineno"> 4161</span>  how2handle_missing_value TEXT,</div> |
| <div class="line"><a name="l04162"></a><span class="lineno"> 4162</span>  max_tree_depth INT,</div> |
| <div class="line"><a name="l04163"></a><span class="lineno"> 4163</span>  sampling_percentage FLOAT8,</div> |
| <div class="line"><a name="l04164"></a><span class="lineno"> 4164</span>  sampling_needed BOOL,</div> |
| <div class="line"><a name="l04165"></a><span class="lineno"> 4165</span>  node_prune_threshold FLOAT8,</div> |
| <div class="line"><a name="l04166"></a><span class="lineno"> 4166</span>  node_split_threshold FLOAT8, </div> |
| <div class="line"><a name="l04167"></a><span class="lineno"> 4167</span>  error_msg TEXT,</div> |
| <div class="line"><a name="l04168"></a><span class="lineno"> 4168</span>  verbosity INT</div> |
| <div class="line"><a name="l04169"></a><span class="lineno"> 4169</span>  ) </div> |
| <div class="line"><a name="l04170"></a><span class="lineno"> 4170</span> RETURNS RECORD AS $$</div> |
| <div class="line"><a name="l04171"></a><span class="lineno"> 4171</span> DECLARE</div> |
| <div class="line"><a name="l04172"></a><span class="lineno"> 4172</span>  table_names TEXT[]; -- 1: encoded table; 2: meta table</div> |
| <div class="line"><a name="l04173"></a><span class="lineno"> 4173</span>  h2hmv_routine_id INT := 1; </div> |
| <div class="line"><a name="l04174"></a><span class="lineno"> 4174</span>  h2hmv_routine_name TEXT;</div> |
| <div class="line"><a name="l04175"></a><span class="lineno"> 4175</span>  n_fids INT;</div> |
| <div class="line"><a name="l04176"></a><span class="lineno"> 4176</span>  curstmt TEXT;</div> |
| <div class="line"><a name="l04177"></a><span class="lineno"> 4177</span>  enc_tree_name TEXT;</div> |
| <div class="line"><a name="l04178"></a><span class="lineno"> 4178</span>  cont_feature_col_names TEXT[];</div> |
| <div class="line"><a name="l04179"></a><span class="lineno"> 4179</span>  feature_name_array TEXT[];</div> |
| <div class="line"><a name="l04180"></a><span class="lineno"> 4180</span>  train_rs MADLIB_SCHEMA.__train_result;</div> |
| <div class="line"><a name="l04181"></a><span class="lineno"> 4181</span> BEGIN</div> |
| <div class="line"><a name="l04182"></a><span class="lineno"> 4182</span>  cont_feature_col_names = MADLIB_SCHEMA.__csvstr_to_array(continuous_feature_names);</div> |
| <div class="line"><a name="l04183"></a><span class="lineno"> 4183</span>  feature_name_array = MADLIB_SCHEMA.__csvstr_to_array(feature_col_names);</div> |
| <div class="line"><a name="l04184"></a><span class="lineno"> 4184</span>  </div> |
| <div class="line"><a name="l04185"></a><span class="lineno"> 4185</span>  -- if the training table is an valid encoded table, then we retrieve</div> |
| <div class="line"><a name="l04186"></a><span class="lineno"> 4186</span>  -- the relevant information from training_info table directly.</div> |
| <div class="line"><a name="l04187"></a><span class="lineno"> 4187</span>  IF (MADLIB_SCHEMA.__is_valid_enc_table(training_table_name)) THEN</div> |
| <div class="line"><a name="l04188"></a><span class="lineno"> 4188</span>  enc_tree_name = MADLIB_SCHEMA.__get_tree_table_name </div> |
| <div class="line"><a name="l04189"></a><span class="lineno"> 4189</span>  (training_table_name);</div> |
| <div class="line"><a name="l04190"></a><span class="lineno"> 4190</span>  table_names[1] = training_table_name; </div> |
| <div class="line"><a name="l04191"></a><span class="lineno"> 4191</span>  table_names[2] = MADLIB_SCHEMA.__get_metatable_name(enc_tree_name);</div> |
| <div class="line"><a name="l04192"></a><span class="lineno"> 4192</span>  h2hmv_routine_name = MADLIB_SCHEMA.__get_routine_name(enc_tree_name);</div> |
| <div class="line"><a name="l04193"></a><span class="lineno"> 4193</span>  IF (h2hmv_routine_name = 'ignore') THEN</div> |
| <div class="line"><a name="l04194"></a><span class="lineno"> 4194</span>  h2hmv_routine_id = 1;</div> |
| <div class="line"><a name="l04195"></a><span class="lineno"> 4195</span>  ELSE</div> |
| <div class="line"><a name="l04196"></a><span class="lineno"> 4196</span>  h2hmv_routine_id = 2;</div> |
| <div class="line"><a name="l04197"></a><span class="lineno"> 4197</span>  END IF;</div> |
| <div class="line"><a name="l04198"></a><span class="lineno"> 4198</span>  </div> |
| <div class="line"><a name="l04199"></a><span class="lineno"> 4199</span>  -- validate the metatable</div> |
| <div class="line"><a name="l04200"></a><span class="lineno"> 4200</span>  PERFORM MADLIB_SCHEMA.__validate_metatable(table_names[2]); </div> |
| <div class="line"><a name="l04201"></a><span class="lineno"> 4201</span> </div> |
| <div class="line"><a name="l04202"></a><span class="lineno"> 4202</span>  n_fids = MADLIB_SCHEMA.__num_of_feature(table_names[2]);</div> |
| <div class="line"><a name="l04203"></a><span class="lineno"> 4203</span>  PERFORM MADLIB_SCHEMA.__assert</div> |
| <div class="line"><a name="l04204"></a><span class="lineno"> 4204</span>  (</div> |
| <div class="line"><a name="l04205"></a><span class="lineno"> 4205</span>  features_per_node IS NULL OR</div> |
| <div class="line"><a name="l04206"></a><span class="lineno"> 4206</span>  n_fids >= features_per_node,</div> |
| <div class="line"><a name="l04207"></a><span class="lineno"> 4207</span>  'the value of features_per_node must be less than or equal to the total number ' ||</div> |
| <div class="line"><a name="l04208"></a><span class="lineno"> 4208</span>  'of features of the training table'</div> |
| <div class="line"><a name="l04209"></a><span class="lineno"> 4209</span>  ); </div> |
| <div class="line"><a name="l04210"></a><span class="lineno"> 4210</span>  -- create tree table and auxiliary tables</div> |
| <div class="line"><a name="l04211"></a><span class="lineno"> 4211</span>  -- so that we can get the schema name of the table </div> |
| <div class="line"><a name="l04212"></a><span class="lineno"> 4212</span>  PERFORM MADLIB_SCHEMA.__create_tree_tables(tree_table_name);</div> |
| <div class="line"><a name="l04213"></a><span class="lineno"> 4213</span>  ELSE</div> |
| <div class="line"><a name="l04214"></a><span class="lineno"> 4214</span>  -- the provided columns must be in the training table</div> |
| <div class="line"><a name="l04215"></a><span class="lineno"> 4215</span>  PERFORM MADLIB_SCHEMA.__check_training_table</div> |
| <div class="line"><a name="l04216"></a><span class="lineno"> 4216</span>  (</div> |
| <div class="line"><a name="l04217"></a><span class="lineno"> 4217</span>  training_table_name,</div> |
| <div class="line"><a name="l04218"></a><span class="lineno"> 4218</span>  cont_feature_col_names, </div> |
| <div class="line"><a name="l04219"></a><span class="lineno"> 4219</span>  feature_name_array, </div> |
| <div class="line"><a name="l04220"></a><span class="lineno"> 4220</span>  id_col_name, </div> |
| <div class="line"><a name="l04221"></a><span class="lineno"> 4221</span>  class_col_name,</div> |
| <div class="line"><a name="l04222"></a><span class="lineno"> 4222</span>  features_per_node</div> |
| <div class="line"><a name="l04223"></a><span class="lineno"> 4223</span>  ); </div> |
| <div class="line"><a name="l04224"></a><span class="lineno"> 4224</span> </div> |
| <div class="line"><a name="l04225"></a><span class="lineno"> 4225</span>  h2hmv_routine_name = btrim(how2handle_missing_value, ' '); </div> |
| <div class="line"><a name="l04226"></a><span class="lineno"> 4226</span>  IF (h2hmv_routine_name = 'ignore') THEN</div> |
| <div class="line"><a name="l04227"></a><span class="lineno"> 4227</span>  h2hmv_routine_id = 1;</div> |
| <div class="line"><a name="l04228"></a><span class="lineno"> 4228</span>  ELSE</div> |
| <div class="line"><a name="l04229"></a><span class="lineno"> 4229</span>  h2hmv_routine_id = 2;</div> |
| <div class="line"><a name="l04230"></a><span class="lineno"> 4230</span>  END IF;</div> |
| <div class="line"><a name="l04231"></a><span class="lineno"> 4231</span>  </div> |
| <div class="line"><a name="l04232"></a><span class="lineno"> 4232</span>  -- create tree table and auxiliary tables</div> |
| <div class="line"><a name="l04233"></a><span class="lineno"> 4233</span>  -- so that we can get the schema name of the table </div> |
| <div class="line"><a name="l04234"></a><span class="lineno"> 4234</span>  PERFORM MADLIB_SCHEMA.__create_tree_tables(tree_table_name);</div> |
| <div class="line"><a name="l04235"></a><span class="lineno"> 4235</span> </div> |
| <div class="line"><a name="l04236"></a><span class="lineno"> 4236</span>  -- encode the training table</div> |
| <div class="line"><a name="l04237"></a><span class="lineno"> 4237</span>  table_names = MADLIB_SCHEMA.__gen_enc_meta_names(tree_table_name, error_msg); </div> |
| <div class="line"><a name="l04238"></a><span class="lineno"> 4238</span>  PERFORM MADLIB_SCHEMA.__encode_table</div> |
| <div class="line"><a name="l04239"></a><span class="lineno"> 4239</span>  (</div> |
| <div class="line"><a name="l04240"></a><span class="lineno"> 4240</span>  training_table_name,</div> |
| <div class="line"><a name="l04241"></a><span class="lineno"> 4241</span>  lower(id_col_name),</div> |
| <div class="line"><a name="l04242"></a><span class="lineno"> 4242</span>  feature_name_array,</div> |
| <div class="line"><a name="l04243"></a><span class="lineno"> 4243</span>  lower(class_col_name),</div> |
| <div class="line"><a name="l04244"></a><span class="lineno"> 4244</span>  cont_feature_col_names,</div> |
| <div class="line"><a name="l04245"></a><span class="lineno"> 4245</span>  table_names[1],</div> |
| <div class="line"><a name="l04246"></a><span class="lineno"> 4246</span>  table_names[2],</div> |
| <div class="line"><a name="l04247"></a><span class="lineno"> 4247</span>  h2hmv_routine_id,</div> |
| <div class="line"><a name="l04248"></a><span class="lineno"> 4248</span>  verbosity</div> |
| <div class="line"><a name="l04249"></a><span class="lineno"> 4249</span>  );</div> |
| <div class="line"><a name="l04250"></a><span class="lineno"> 4250</span>  n_fids = MADLIB_SCHEMA.__num_of_feature(table_names[2]);</div> |
| <div class="line"><a name="l04251"></a><span class="lineno"> 4251</span>  END IF;</div> |
| <div class="line"><a name="l04252"></a><span class="lineno"> 4252</span>  </div> |
| <div class="line"><a name="l04253"></a><span class="lineno"> 4253</span>  IF (sampling_needed) THEN</div> |
| <div class="line"><a name="l04254"></a><span class="lineno"> 4254</span>  IF (features_per_node IS NULL) THEN</div> |
| <div class="line"><a name="l04255"></a><span class="lineno"> 4255</span>  n_fids = round(sqrt(n_fids) - 0.5)::INT + 1;</div> |
| <div class="line"><a name="l04256"></a><span class="lineno"> 4256</span>  ELSE</div> |
| <div class="line"><a name="l04257"></a><span class="lineno"> 4257</span>  n_fids = features_per_node;</div> |
| <div class="line"><a name="l04258"></a><span class="lineno"> 4258</span>  END IF;</div> |
| <div class="line"><a name="l04259"></a><span class="lineno"> 4259</span>  END IF;</div> |
| <div class="line"><a name="l04260"></a><span class="lineno"> 4260</span> </div> |
| <div class="line"><a name="l04261"></a><span class="lineno"> 4261</span>  IF (verbosity > 0) THEN</div> |
| <div class="line"><a name="l04262"></a><span class="lineno"> 4262</span>  RAISE INFO 'features_per_node: %', n_fids;</div> |
| <div class="line"><a name="l04263"></a><span class="lineno"> 4263</span>  END IF;</div> |
| <div class="line"><a name="l04264"></a><span class="lineno"> 4264</span>  </div> |
| <div class="line"><a name="l04265"></a><span class="lineno"> 4265</span>  -- insert data to the training_info table</div> |
| <div class="line"><a name="l04266"></a><span class="lineno"> 4266</span>  PERFORM MADLIB_SCHEMA.__insert_into_traininginfo</div> |
| <div class="line"><a name="l04267"></a><span class="lineno"> 4267</span>  (</div> |
| <div class="line"><a name="l04268"></a><span class="lineno"> 4268</span>  dt_algo_name,</div> |
| <div class="line"><a name="l04269"></a><span class="lineno"> 4269</span>  tree_table_name,</div> |
| <div class="line"><a name="l04270"></a><span class="lineno"> 4270</span>  training_table_name,</div> |
| <div class="line"><a name="l04271"></a><span class="lineno"> 4271</span>  table_names[2],</div> |
| <div class="line"><a name="l04272"></a><span class="lineno"> 4272</span>  table_names[1],</div> |
| <div class="line"><a name="l04273"></a><span class="lineno"> 4273</span>  validation_table_name,</div> |
| <div class="line"><a name="l04274"></a><span class="lineno"> 4274</span>  h2hmv_routine_name,</div> |
| <div class="line"><a name="l04275"></a><span class="lineno"> 4275</span>  split_criterion,</div> |
| <div class="line"><a name="l04276"></a><span class="lineno"> 4276</span>  sampling_percentage,</div> |
| <div class="line"><a name="l04277"></a><span class="lineno"> 4277</span>  n_fids,</div> |
| <div class="line"><a name="l04278"></a><span class="lineno"> 4278</span>  num_trees</div> |
| <div class="line"><a name="l04279"></a><span class="lineno"> 4279</span>  );</div> |
| <div class="line"><a name="l04280"></a><span class="lineno"> 4280</span>  </div> |
| <div class="line"><a name="l04281"></a><span class="lineno"> 4281</span>  -- call the tree grow engine</div> |
| <div class="line"><a name="l04282"></a><span class="lineno"> 4282</span>  train_rs = MADLIB_SCHEMA.__train_tree</div> |
| <div class="line"><a name="l04283"></a><span class="lineno"> 4283</span>  (</div> |
| <div class="line"><a name="l04284"></a><span class="lineno"> 4284</span>  split_criterion,</div> |
| <div class="line"><a name="l04285"></a><span class="lineno"> 4285</span>  num_trees,</div> |
| <div class="line"><a name="l04286"></a><span class="lineno"> 4286</span>  n_fids ,</div> |
| <div class="line"><a name="l04287"></a><span class="lineno"> 4287</span>  table_names[1],</div> |
| <div class="line"><a name="l04288"></a><span class="lineno"> 4288</span>  table_names[2],</div> |
| <div class="line"><a name="l04289"></a><span class="lineno"> 4289</span>  tree_table_name,</div> |
| <div class="line"><a name="l04290"></a><span class="lineno"> 4290</span>  validation_table_name, </div> |
| <div class="line"><a name="l04291"></a><span class="lineno"> 4291</span>  'id', </div> |
| <div class="line"><a name="l04292"></a><span class="lineno"> 4292</span>  'class', </div> |
| <div class="line"><a name="l04293"></a><span class="lineno"> 4293</span>  confidence_level, </div> |
| <div class="line"><a name="l04294"></a><span class="lineno"> 4294</span>  max_tree_depth, </div> |
| <div class="line"><a name="l04295"></a><span class="lineno"> 4295</span>  sampling_percentage,</div> |
| <div class="line"><a name="l04296"></a><span class="lineno"> 4296</span>  node_prune_threshold,</div> |
| <div class="line"><a name="l04297"></a><span class="lineno"> 4297</span>  node_split_threshold,</div> |
| <div class="line"><a name="l04298"></a><span class="lineno"> 4298</span>  sampling_needed,</div> |
| <div class="line"><a name="l04299"></a><span class="lineno"> 4299</span>  h2hmv_routine_id,</div> |
| <div class="line"><a name="l04300"></a><span class="lineno"> 4300</span>  verbosity</div> |
| <div class="line"><a name="l04301"></a><span class="lineno"> 4301</span>  );</div> |
| <div class="line"><a name="l04302"></a><span class="lineno"> 4302</span> </div> |
| <div class="line"><a name="l04303"></a><span class="lineno"> 4303</span>  RETURN train_rs;</div> |
| <div class="line"><a name="l04304"></a><span class="lineno"> 4304</span> END</div> |
| <div class="line"><a name="l04305"></a><span class="lineno"> 4305</span> $$ LANGUAGE PLPGSQL STABLE;</div> |
| </div><!-- fragment --></div><!-- contents --> |
| </div><!-- doc-content --> |
| <!-- start footer part --> |
| <div id="nav-path" class="navpath"><!-- id is needed for treeview function! --> |
| <ul> |
| <li class="navelem"><a class="el" href="dir_704eb8350b43e1ca74c0f90ed1ba450e.html">methods</a></li><li class="navelem"><a class="el" href="dir_fbc4f2259ae1d6f6cc75298ebbd15532.html">cart</a></li><li class="navelem"><a class="el" href="dir_42a199e31e82b0c41cf7961a25e597db.html">src</a></li><li class="navelem"><a class="el" href="dir_64bd18b6b0e4b6a9c2cd2ca1d5a05b4c.html">pg_gp</a></li><li class="navelem"><a class="el" href="dt_8sql__in.html">dt.sql_in</a></li> |
| <li class="footer">Generated on Tue Sep 10 2013 15:48:04 for MADlib by |
| <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.4 </li> |
| </ul> |
| </div> |
| </body> |
| </html> |
