141 lines
41 KiB
HTML
141 lines
41 KiB
HTML
<!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.11"/>
|
|
<title>mlpack: src/mlpack/methods/decision_tree/decision_tree.hpp 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="search/search.css" rel="stylesheet" type="text/css"/>
|
|
<script type="text/javascript" src="search/searchdata.js"></script>
|
|
<script type="text/javascript" src="search/search.js"></script>
|
|
<script type="text/javascript">
|
|
$(document).ready(function() { init_search(); });
|
|
</script>
|
|
<link href="doxygen.css" rel="stylesheet" type="text/css" />
|
|
<link href="extra-stylesheet.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 id="projectalign" style="padding-left: 0.5em;">
|
|
<div id="projectname">mlpack
|
|
 <span id="projectnumber">master</span>
|
|
</div>
|
|
</td>
|
|
</tr>
|
|
</tbody>
|
|
</table>
|
|
</div>
|
|
<!-- end header part -->
|
|
<!-- Generated by Doxygen 1.8.11 -->
|
|
<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="pages.html"><span>Related Pages</span></a></li>
|
|
<li><a href="namespaces.html"><span>Namespaces</span></a></li>
|
|
<li><a href="annotated.html"><span>Classes</span></a></li>
|
|
<li class="current"><a href="files.html"><span>Files</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 id="navrow2" class="tabs2">
|
|
<ul class="tablist">
|
|
<li><a href="files.html"><span>File List</span></a></li>
|
|
<li><a href="globals.html"><span>File Members</span></a></li>
|
|
</ul>
|
|
</div>
|
|
<!-- window showing the filter options -->
|
|
<div id="MSearchSelectWindow"
|
|
onmouseover="return searchBox.OnSearchSelectShow()"
|
|
onmouseout="return searchBox.OnSearchSelectHide()"
|
|
onkeydown="return searchBox.OnSearchSelectKey(event)">
|
|
</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 id="nav-path" class="navpath">
|
|
<ul>
|
|
<li class="navelem"><a class="el" href="dir_68267d1309a1af8e8297ef4c3efbcdba.html">src</a></li><li class="navelem"><a class="el" href="dir_44a936ef30fc126706c6284a2fc0c990.html">mlpack</a></li><li class="navelem"><a class="el" href="dir_3cfbf900288ec2fe8dd2c2c8482f216e.html">methods</a></li><li class="navelem"><a class="el" href="dir_f295b20b1dd4e5e0a64437251c55c9ac.html">decision_tree</a></li> </ul>
|
|
</div>
|
|
</div><!-- top -->
|
|
<div class="header">
|
|
<div class="headertitle">
|
|
<div class="title">decision_tree.hpp</div> </div>
|
|
</div><!--header-->
|
|
<div class="contents">
|
|
<a href="decision__tree_8hpp.html">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span> </div><div class="line"><a name="l00008"></a><span class="lineno"> 8</span> <span class="preprocessor">#ifndef MLPACK_METHODS_DECISION_TREE_DECISION_TREE_HPP</span></div><div class="line"><a name="l00009"></a><span class="lineno"> 9</span> <span class="preprocessor">#define MLPACK_METHODS_DECISION_TREE_DECISION_TREE_HPP</span></div><div class="line"><a name="l00010"></a><span class="lineno"> 10</span> </div><div class="line"><a name="l00011"></a><span class="lineno"> 11</span> <span class="preprocessor">#include <<a class="code" href="prereqs_8hpp.html">mlpack/prereqs.hpp</a>></span></div><div class="line"><a name="l00012"></a><span class="lineno"> 12</span> <span class="preprocessor">#include "<a class="code" href="gini__gain_8hpp.html">gini_gain.hpp</a>"</span></div><div class="line"><a name="l00013"></a><span class="lineno"> 13</span> <span class="preprocessor">#include "<a class="code" href="best__binary__numeric__split_8hpp.html">best_binary_numeric_split.hpp</a>"</span></div><div class="line"><a name="l00014"></a><span class="lineno"> 14</span> <span class="preprocessor">#include "<a class="code" href="all__categorical__split_8hpp.html">all_categorical_split.hpp</a>"</span></div><div class="line"><a name="l00015"></a><span class="lineno"> 15</span> </div><div class="line"><a name="l00016"></a><span class="lineno"> 16</span> <span class="keyword">namespace </span><a class="code" href="namespacemlpack.html">mlpack</a> {</div><div class="line"><a name="l00017"></a><span class="lineno"> 17</span> <span class="keyword">namespace </span>tree {</div><div class="line"><a name="l00018"></a><span class="lineno"> 18</span> </div><div class="line"><a name="l00026"></a><span class="lineno"> 26</span> <span class="keyword">template</span><<span class="keyword">typename</span> FitnessFunction = GiniGain,</div><div class="line"><a name="l00027"></a><span class="lineno"> 27</span>  <span class="keyword">template</span><<span class="keyword">typename</span>> <span class="keyword">class </span>NumericSplitType = BestBinaryNumericSplit,</div><div class="line"><a name="l00028"></a><span class="lineno"> 28</span>  <span class="keyword">template</span><<span class="keyword">typename</span>> <span class="keyword">class </span>CategoricalSplitType = AllCategoricalSplit,</div><div class="line"><a name="l00029"></a><span class="lineno"> 29</span>  <span class="keyword">typename</span> ElemType = double,</div><div class="line"><a name="l00030"></a><span class="lineno"> 30</span>  <span class="keywordtype">bool</span> NoRecursion = <span class="keyword">false</span>></div><div class="line"><a name="l00031"></a><span class="lineno"><a class="line" href="classmlpack_1_1tree_1_1DecisionTree.html"> 31</a></span> <span class="keyword">class </span><a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html">DecisionTree</a> :</div><div class="line"><a name="l00032"></a><span class="lineno"> 32</span>  <span class="keyword">public</span> NumericSplitType<FitnessFunction>::template</div><div class="line"><a name="l00033"></a><span class="lineno"> 33</span>  AuxiliarySplitInfo<ElemType>,</div><div class="line"><a name="l00034"></a><span class="lineno"> 34</span>  <span class="keyword">public</span> CategoricalSplitType<FitnessFunction>::template</div><div class="line"><a name="l00035"></a><span class="lineno"> 35</span>  AuxiliarySplitInfo<ElemType></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>  <span class="keyword">public</span>:</div><div class="line"><a name="l00039"></a><span class="lineno"><a class="line" href="classmlpack_1_1tree_1_1DecisionTree.html#aaa0076abd32aae8eb6bc04d98b843faf"> 39</a></span>  <span class="keyword">typedef</span> NumericSplitType<FitnessFunction> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#aaa0076abd32aae8eb6bc04d98b843faf">NumericSplit</a>;</div><div class="line"><a name="l00041"></a><span class="lineno"><a class="line" href="classmlpack_1_1tree_1_1DecisionTree.html#a47c926608af4037637e7a71b2788c780"> 41</a></span>  <span class="keyword">typedef</span> CategoricalSplitType<FitnessFunction> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a47c926608af4037637e7a71b2788c780">CategoricalSplit</a>;</div><div class="line"><a name="l00042"></a><span class="lineno"> 42</span> </div><div class="line"><a name="l00055"></a><span class="lineno"> 55</span>  <span class="keyword">template</span><<span class="keyword">typename</span> MatType></div><div class="line"><a name="l00056"></a><span class="lineno"> 56</span>  <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a8895e26a965b754ac13b1aee3babfcfc">DecisionTree</a>(<span class="keyword">const</span> MatType& data,</div><div class="line"><a name="l00057"></a><span class="lineno"> 57</span>  <span class="keyword">const</span> <a class="code" href="classmlpack_1_1data_1_1DatasetMapper.html">data::DatasetInfo</a>& datasetInfo,</div><div class="line"><a name="l00058"></a><span class="lineno"> 58</span>  <span class="keyword">const</span> arma::Row<size_t>& labels,</div><div class="line"><a name="l00059"></a><span class="lineno"> 59</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span> numClasses,</div><div class="line"><a name="l00060"></a><span class="lineno"> 60</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span> minimumLeafSize = 10);</div><div class="line"><a name="l00061"></a><span class="lineno"> 61</span> </div><div class="line"><a name="l00073"></a><span class="lineno"> 73</span>  <span class="keyword">template</span><<span class="keyword">typename</span> MatType></div><div class="line"><a name="l00074"></a><span class="lineno"> 74</span>  <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a8895e26a965b754ac13b1aee3babfcfc">DecisionTree</a>(<span class="keyword">const</span> MatType& data,</div><div class="line"><a name="l00075"></a><span class="lineno"> 75</span>  <span class="keyword">const</span> arma::Row<size_t>& labels,</div><div class="line"><a name="l00076"></a><span class="lineno"> 76</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span> numClasses,</div><div class="line"><a name="l00077"></a><span class="lineno"> 77</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span> minimumLeafSize = 10);</div><div class="line"><a name="l00078"></a><span class="lineno"> 78</span> </div><div class="line"><a name="l00085"></a><span class="lineno"> 85</span>  <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a8895e26a965b754ac13b1aee3babfcfc">DecisionTree</a>(<span class="keyword">const</span> <span class="keywordtype">size_t</span> numClasses = 1);</div><div class="line"><a name="l00086"></a><span class="lineno"> 86</span> </div><div class="line"><a name="l00093"></a><span class="lineno"> 93</span>  <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a8895e26a965b754ac13b1aee3babfcfc">DecisionTree</a>(<span class="keyword">const</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html">DecisionTree</a>& other);</div><div class="line"><a name="l00094"></a><span class="lineno"> 94</span> </div><div class="line"><a name="l00100"></a><span class="lineno"> 100</span>  <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a8895e26a965b754ac13b1aee3babfcfc">DecisionTree</a>(<a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html">DecisionTree</a>&& other);</div><div class="line"><a name="l00101"></a><span class="lineno"> 101</span> </div><div class="line"><a name="l00108"></a><span class="lineno"> 108</span>  <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html">DecisionTree</a>& <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#af9a0efae345916db75d4de55ee800d0d">operator=</a>(<span class="keyword">const</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html">DecisionTree</a>& other);</div><div class="line"><a name="l00109"></a><span class="lineno"> 109</span> </div><div class="line"><a name="l00115"></a><span class="lineno"> 115</span>  <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html">DecisionTree</a>& <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#af9a0efae345916db75d4de55ee800d0d">operator=</a>(<a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html">DecisionTree</a>&& other);</div><div class="line"><a name="l00116"></a><span class="lineno"> 116</span> </div><div class="line"><a name="l00120"></a><span class="lineno"> 120</span>  <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a7f944c4f36862ab630f8723c5d3f8ee9">~DecisionTree</a>();</div><div class="line"><a name="l00121"></a><span class="lineno"> 121</span> </div><div class="line"><a name="l00134"></a><span class="lineno"> 134</span>  <span class="keyword">template</span><<span class="keyword">typename</span> MatType></div><div class="line"><a name="l00135"></a><span class="lineno"> 135</span>  <span class="keywordtype">void</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a9a56744b932b6e6c78b4975c9e355896">Train</a>(<span class="keyword">const</span> MatType& data,</div><div class="line"><a name="l00136"></a><span class="lineno"> 136</span>  <span class="keyword">const</span> <a class="code" href="classmlpack_1_1data_1_1DatasetMapper.html">data::DatasetInfo</a>& datasetInfo,</div><div class="line"><a name="l00137"></a><span class="lineno"> 137</span>  <span class="keyword">const</span> arma::Row<size_t>& labels,</div><div class="line"><a name="l00138"></a><span class="lineno"> 138</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span> numClasses,</div><div class="line"><a name="l00139"></a><span class="lineno"> 139</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span> minimumLeafSize = 10);</div><div class="line"><a name="l00140"></a><span class="lineno"> 140</span> </div><div class="line"><a name="l00152"></a><span class="lineno"> 152</span>  <span class="keyword">template</span><<span class="keyword">typename</span> MatType></div><div class="line"><a name="l00153"></a><span class="lineno"> 153</span>  <span class="keywordtype">void</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a9a56744b932b6e6c78b4975c9e355896">Train</a>(<span class="keyword">const</span> MatType& data,</div><div class="line"><a name="l00154"></a><span class="lineno"> 154</span>  <span class="keyword">const</span> arma::Row<size_t>& labels,</div><div class="line"><a name="l00155"></a><span class="lineno"> 155</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span> numClasses,</div><div class="line"><a name="l00156"></a><span class="lineno"> 156</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span> minimumLeafSize = 10);</div><div class="line"><a name="l00157"></a><span class="lineno"> 157</span> </div><div class="line"><a name="l00164"></a><span class="lineno"> 164</span>  <span class="keyword">template</span><<span class="keyword">typename</span> VecType></div><div class="line"><a name="l00165"></a><span class="lineno"> 165</span>  <span class="keywordtype">size_t</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a81d8d5e76ac1b6c58341a007290847b6">Classify</a>(<span class="keyword">const</span> VecType& point) <span class="keyword">const</span>;</div><div class="line"><a name="l00166"></a><span class="lineno"> 166</span> </div><div class="line"><a name="l00176"></a><span class="lineno"> 176</span>  <span class="keyword">template</span><<span class="keyword">typename</span> VecType></div><div class="line"><a name="l00177"></a><span class="lineno"> 177</span>  <span class="keywordtype">void</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a81d8d5e76ac1b6c58341a007290847b6">Classify</a>(<span class="keyword">const</span> VecType& point,</div><div class="line"><a name="l00178"></a><span class="lineno"> 178</span>  <span class="keywordtype">size_t</span>& prediction,</div><div class="line"><a name="l00179"></a><span class="lineno"> 179</span>  arma::vec& probabilities) <span class="keyword">const</span>;</div><div class="line"><a name="l00180"></a><span class="lineno"> 180</span> </div><div class="line"><a name="l00188"></a><span class="lineno"> 188</span>  <span class="keyword">template</span><<span class="keyword">typename</span> MatType></div><div class="line"><a name="l00189"></a><span class="lineno"> 189</span>  <span class="keywordtype">void</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a81d8d5e76ac1b6c58341a007290847b6">Classify</a>(<span class="keyword">const</span> MatType& data,</div><div class="line"><a name="l00190"></a><span class="lineno"> 190</span>  arma::Row<size_t>& predictions) <span class="keyword">const</span>;</div><div class="line"><a name="l00191"></a><span class="lineno"> 191</span> </div><div class="line"><a name="l00202"></a><span class="lineno"> 202</span>  <span class="keyword">template</span><<span class="keyword">typename</span> MatType></div><div class="line"><a name="l00203"></a><span class="lineno"> 203</span>  <span class="keywordtype">void</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a81d8d5e76ac1b6c58341a007290847b6">Classify</a>(<span class="keyword">const</span> MatType& data,</div><div class="line"><a name="l00204"></a><span class="lineno"> 204</span>  arma::Row<size_t>& predictions,</div><div class="line"><a name="l00205"></a><span class="lineno"> 205</span>  arma::mat& probabilities) <span class="keyword">const</span>;</div><div class="line"><a name="l00206"></a><span class="lineno"> 206</span> </div><div class="line"><a name="l00210"></a><span class="lineno"> 210</span>  <span class="keyword">template</span><<span class="keyword">typename</span> Archive></div><div class="line"><a name="l00211"></a><span class="lineno"> 211</span>  <span class="keywordtype">void</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a4372a14aa83472be410486c299990816">Serialize</a>(Archive& ar, <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> <span class="comment">/* version */</span>);</div><div class="line"><a name="l00212"></a><span class="lineno"> 212</span> </div><div class="line"><a name="l00214"></a><span class="lineno"><a class="line" href="classmlpack_1_1tree_1_1DecisionTree.html#a51b2801e7255ae796517583a04cf41cc"> 214</a></span>  <span class="keywordtype">size_t</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a51b2801e7255ae796517583a04cf41cc">NumChildren</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a3743ab3245d148f78817f5d1b812f6e8">children</a>.size(); }</div><div class="line"><a name="l00215"></a><span class="lineno"> 215</span> </div><div class="line"><a name="l00217"></a><span class="lineno"><a class="line" href="classmlpack_1_1tree_1_1DecisionTree.html#abb477016c1ef13959ab35b025c6fa97c"> 217</a></span>  <span class="keyword">const</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html">DecisionTree</a>& <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#abb477016c1ef13959ab35b025c6fa97c">Child</a>(<span class="keyword">const</span> <span class="keywordtype">size_t</span> i)<span class="keyword"> const </span>{ <span class="keywordflow">return</span> *<a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a3743ab3245d148f78817f5d1b812f6e8">children</a>[i]; }</div><div class="line"><a name="l00219"></a><span class="lineno"><a class="line" href="classmlpack_1_1tree_1_1DecisionTree.html#a5d24d8ce5ccadbc604f4d1bf41e87449"> 219</a></span>  <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html">DecisionTree</a>& <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a5d24d8ce5ccadbc604f4d1bf41e87449">Child</a>(<span class="keyword">const</span> <span class="keywordtype">size_t</span> i) { <span class="keywordflow">return</span> *<a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a3743ab3245d148f78817f5d1b812f6e8">children</a>[i]; }</div><div class="line"><a name="l00220"></a><span class="lineno"> 220</span> </div><div class="line"><a name="l00228"></a><span class="lineno"> 228</span>  <span class="keyword">template</span><<span class="keyword">typename</span> VecType></div><div class="line"><a name="l00229"></a><span class="lineno"> 229</span>  <span class="keywordtype">size_t</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a09ecf7d2d8a1ff6e74d634c1153909c7">CalculateDirection</a>(<span class="keyword">const</span> VecType& point) <span class="keyword">const</span>;</div><div class="line"><a name="l00230"></a><span class="lineno"> 230</span> </div><div class="line"><a name="l00231"></a><span class="lineno"> 231</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00233"></a><span class="lineno"><a class="line" href="classmlpack_1_1tree_1_1DecisionTree.html#a3743ab3245d148f78817f5d1b812f6e8"> 233</a></span>  std::vector<DecisionTree*> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a3743ab3245d148f78817f5d1b812f6e8">children</a>;</div><div class="line"><a name="l00235"></a><span class="lineno"><a class="line" href="classmlpack_1_1tree_1_1DecisionTree.html#acb83dcda9656c66d0bff4f347d5e0a98"> 235</a></span>  <span class="keywordtype">size_t</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#acb83dcda9656c66d0bff4f347d5e0a98">splitDimension</a>;</div><div class="line"><a name="l00238"></a><span class="lineno"><a class="line" href="classmlpack_1_1tree_1_1DecisionTree.html#aa8550c35ef6755b61d2467aa1347f4a6"> 238</a></span>  <span class="keywordtype">size_t</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#aa8550c35ef6755b61d2467aa1347f4a6">dimensionTypeOrMajorityClass</a>;</div><div class="line"><a name="l00246"></a><span class="lineno"><a class="line" href="classmlpack_1_1tree_1_1DecisionTree.html#a9b014767c5983cc9dcc3564e6167e2d1"> 246</a></span>  arma::vec <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a9b014767c5983cc9dcc3564e6167e2d1">classProbabilities</a>;</div><div class="line"><a name="l00247"></a><span class="lineno"> 247</span> </div><div class="line"><a name="l00251"></a><span class="lineno"> 251</span>  <span class="keyword">typedef</span> <span class="keyword">typename</span> NumericSplit::template AuxiliarySplitInfo<ElemType></div><div class="line"><a name="l00252"></a><span class="lineno"><a class="line" href="classmlpack_1_1tree_1_1DecisionTree.html#aaa53d2150636d2e7d4dfde8030e7377d"> 252</a></span>  <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#aaa53d2150636d2e7d4dfde8030e7377d">NumericAuxiliarySplitInfo</a>;</div><div class="line"><a name="l00253"></a><span class="lineno"> 253</span>  <span class="keyword">typedef</span> <span class="keyword">typename</span> CategoricalSplit::template AuxiliarySplitInfo<ElemType></div><div class="line"><a name="l00254"></a><span class="lineno"><a class="line" href="classmlpack_1_1tree_1_1DecisionTree.html#a8f71e2e71f5b875fe1b6b6e3b2286358"> 254</a></span>  <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#a8f71e2e71f5b875fe1b6b6e3b2286358">CategoricalAuxiliarySplitInfo</a>;</div><div class="line"><a name="l00255"></a><span class="lineno"> 255</span> </div><div class="line"><a name="l00259"></a><span class="lineno"> 259</span>  <span class="keyword">template</span><<span class="keyword">typename</span> RowType></div><div class="line"><a name="l00260"></a><span class="lineno"> 260</span>  <span class="keywordtype">void</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html#abaae8925e6735cfeb99a4ab5ebcf4ca3">CalculateClassProbabilities</a>(<span class="keyword">const</span> RowType& labels,</div><div class="line"><a name="l00261"></a><span class="lineno"> 261</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span> numClasses);</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> </div><div class="line"><a name="l00267"></a><span class="lineno"> 267</span> <span class="keyword">template</span><<span class="keyword">typename</span> FitnessFunction = <a class="code" href="classmlpack_1_1tree_1_1GiniGain.html">GiniGain</a>,</div><div class="line"><a name="l00268"></a><span class="lineno"> 268</span>  <span class="keyword">template</span><<span class="keyword">typename</span>> <span class="keyword">class </span>NumericSplitType = <a class="code" href="classmlpack_1_1tree_1_1BestBinaryNumericSplit.html">BestBinaryNumericSplit</a>,</div><div class="line"><a name="l00269"></a><span class="lineno"> 269</span>  <span class="keyword">template</span><<span class="keyword">typename</span>> <span class="keyword">class </span>CategoricalSplitType = <a class="code" href="classmlpack_1_1tree_1_1AllCategoricalSplit.html">AllCategoricalSplit</a>,</div><div class="line"><a name="l00270"></a><span class="lineno"> 270</span>  <span class="keyword">typename</span> ElemType = <span class="keywordtype">double</span>></div><div class="line"><a name="l00271"></a><span class="lineno"> 271</span> <span class="keyword">using</span> <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html">DecisionStump</a> = <a class="code" href="classmlpack_1_1tree_1_1DecisionTree.html">DecisionTree</a><FitnessFunction,</div><div class="line"><a name="l00272"></a><span class="lineno"> 272</span>  NumericSplitType,</div><div class="line"><a name="l00273"></a><span class="lineno"> 273</span>  CategoricalSplitType,</div><div class="line"><a name="l00274"></a><span class="lineno"> 274</span>  ElemType,</div><div class="line"><a name="l00275"></a><span class="lineno"><a class="line" href="namespacemlpack_1_1tree.html#adb29d1f69355144f569e4d820a04242a"> 275</a></span>  <span class="keyword">false</span>>;</div><div class="line"><a name="l00276"></a><span class="lineno"> 276</span> </div><div class="line"><a name="l00277"></a><span class="lineno"> 277</span> } <span class="comment">// namespace tree</span></div><div class="line"><a name="l00278"></a><span class="lineno"> 278</span> } <span class="comment">// namespace mlpack</span></div><div class="line"><a name="l00279"></a><span class="lineno"> 279</span> </div><div class="line"><a name="l00280"></a><span class="lineno"> 280</span> <span class="comment">// Include implementation.</span></div><div class="line"><a name="l00281"></a><span class="lineno"> 281</span> <span class="preprocessor">#include "decision_tree_impl.hpp"</span></div><div class="line"><a name="l00282"></a><span class="lineno"> 282</span> </div><div class="line"><a name="l00283"></a><span class="lineno"> 283</span> <span class="preprocessor">#endif</span></div><div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_a09ecf7d2d8a1ff6e74d634c1153909c7"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#a09ecf7d2d8a1ff6e74d634c1153909c7">mlpack::tree::DecisionTree::CalculateDirection</a></div><div class="ttdeci">size_t CalculateDirection(const VecType &point) const </div><div class="ttdoc">Given a point and that this node is not a leaf, calculate the index of the child node this point woul...</div></div>
|
|
<div class="ttc" id="classmlpack_1_1data_1_1DatasetMapper_html"><div class="ttname"><a href="classmlpack_1_1data_1_1DatasetMapper.html">mlpack::data::DatasetMapper</a></div><div class="ttdoc">Auxiliary information for a dataset, including mappings to/from strings and the datatype of each dime...</div><div class="ttdef"><b>Definition:</b> <a href="dataset__mapper_8hpp_source.html#l00036">dataset_mapper.hpp:36</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1BestBinaryNumericSplit_html"><div class="ttname"><a href="classmlpack_1_1tree_1_1BestBinaryNumericSplit.html">mlpack::tree::BestBinaryNumericSplit</a></div><div class="ttdoc">The BestBinaryNumericSplit is a splitting function for decision trees that will exhaustively search a...</div><div class="ttdef"><b>Definition:</b> <a href="best__binary__numeric__split_8hpp_source.html#l00022">best_binary_numeric_split.hpp:22</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_abaae8925e6735cfeb99a4ab5ebcf4ca3"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#abaae8925e6735cfeb99a4ab5ebcf4ca3">mlpack::tree::DecisionTree::CalculateClassProbabilities</a></div><div class="ttdeci">void CalculateClassProbabilities(const RowType &labels, const size_t numClasses)</div><div class="ttdoc">Calculate the class probabilities of the given labels. </div></div>
|
|
<div class="ttc" id="namespacemlpack_html"><div class="ttname"><a href="namespacemlpack.html">mlpack</a></div><div class="ttdoc">Linear algebra utility functions, generally performed on matrices or vectors. </div><div class="ttdef"><b>Definition:</b> <a href="binarize_8hpp_source.html#l00018">binarize.hpp:18</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_a47c926608af4037637e7a71b2788c780"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#a47c926608af4037637e7a71b2788c780">mlpack::tree::DecisionTree::CategoricalSplit</a></div><div class="ttdeci">CategoricalSplitType< FitnessFunction > CategoricalSplit</div><div class="ttdoc">Allow access to the categorical split type. </div><div class="ttdef"><b>Definition:</b> <a href="decision__tree_8hpp_source.html#l00041">decision_tree.hpp:41</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html">mlpack::tree::DecisionTree</a></div><div class="ttdoc">This class implements a generic decision tree learner. </div><div class="ttdef"><b>Definition:</b> <a href="decision__tree_8hpp_source.html#l00031">decision_tree.hpp:31</a></div></div>
|
|
<div class="ttc" id="prereqs_8hpp_html"><div class="ttname"><a href="prereqs_8hpp.html">prereqs.hpp</a></div><div class="ttdoc">The core includes that mlpack expects; standard C++ includes and Armadillo. </div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_aaa0076abd32aae8eb6bc04d98b843faf"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#aaa0076abd32aae8eb6bc04d98b843faf">mlpack::tree::DecisionTree::NumericSplit</a></div><div class="ttdeci">NumericSplitType< FitnessFunction > NumericSplit</div><div class="ttdoc">Allow access to the numeric split type. </div><div class="ttdef"><b>Definition:</b> <a href="decision__tree_8hpp_source.html#l00039">decision_tree.hpp:39</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_a81d8d5e76ac1b6c58341a007290847b6"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#a81d8d5e76ac1b6c58341a007290847b6">mlpack::tree::DecisionTree::Classify</a></div><div class="ttdeci">size_t Classify(const VecType &point) const </div><div class="ttdoc">Classify the given point, using the entire tree. </div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_abb477016c1ef13959ab35b025c6fa97c"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#abb477016c1ef13959ab35b025c6fa97c">mlpack::tree::DecisionTree::Child</a></div><div class="ttdeci">const DecisionTree & Child(const size_t i) const </div><div class="ttdoc">Get the child of the given index. </div><div class="ttdef"><b>Definition:</b> <a href="decision__tree_8hpp_source.html#l00217">decision_tree.hpp:217</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1AllCategoricalSplit_html"><div class="ttname"><a href="classmlpack_1_1tree_1_1AllCategoricalSplit.html">mlpack::tree::AllCategoricalSplit</a></div><div class="ttdoc">The AllCategoricalSplit is a splitting function that will split categorical features into many childr...</div><div class="ttdef"><b>Definition:</b> <a href="all__categorical__split_8hpp_source.html#l00023">all_categorical_split.hpp:23</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_a51b2801e7255ae796517583a04cf41cc"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#a51b2801e7255ae796517583a04cf41cc">mlpack::tree::DecisionTree::NumChildren</a></div><div class="ttdeci">size_t NumChildren() const </div><div class="ttdoc">Get the number of children. </div><div class="ttdef"><b>Definition:</b> <a href="decision__tree_8hpp_source.html#l00214">decision_tree.hpp:214</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_a9b014767c5983cc9dcc3564e6167e2d1"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#a9b014767c5983cc9dcc3564e6167e2d1">mlpack::tree::DecisionTree::classProbabilities</a></div><div class="ttdeci">arma::vec classProbabilities</div><div class="ttdoc">This vector may hold different things. </div><div class="ttdef"><b>Definition:</b> <a href="decision__tree_8hpp_source.html#l00246">decision_tree.hpp:246</a></div></div>
|
|
<div class="ttc" id="best__binary__numeric__split_8hpp_html"><div class="ttname"><a href="best__binary__numeric__split_8hpp.html">best_binary_numeric_split.hpp</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_a3743ab3245d148f78817f5d1b812f6e8"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#a3743ab3245d148f78817f5d1b812f6e8">mlpack::tree::DecisionTree::children</a></div><div class="ttdeci">std::vector< DecisionTree * > children</div><div class="ttdoc">The vector of children. </div><div class="ttdef"><b>Definition:</b> <a href="decision__tree_8hpp_source.html#l00233">decision_tree.hpp:233</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_a5d24d8ce5ccadbc604f4d1bf41e87449"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#a5d24d8ce5ccadbc604f4d1bf41e87449">mlpack::tree::DecisionTree::Child</a></div><div class="ttdeci">DecisionTree & Child(const size_t i)</div><div class="ttdoc">Modify the child of the given index (be careful!). </div><div class="ttdef"><b>Definition:</b> <a href="decision__tree_8hpp_source.html#l00219">decision_tree.hpp:219</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_acb83dcda9656c66d0bff4f347d5e0a98"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#acb83dcda9656c66d0bff4f347d5e0a98">mlpack::tree::DecisionTree::splitDimension</a></div><div class="ttdeci">size_t splitDimension</div><div class="ttdoc">The dimension this node splits on. </div><div class="ttdef"><b>Definition:</b> <a href="decision__tree_8hpp_source.html#l00235">decision_tree.hpp:235</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1GiniGain_html"><div class="ttname"><a href="classmlpack_1_1tree_1_1GiniGain.html">mlpack::tree::GiniGain</a></div><div class="ttdoc">The Gini gain, a measure of set purity usable as a fitness function (FitnessFunction) for decision tr...</div><div class="ttdef"><b>Definition:</b> <a href="gini__gain_8hpp_source.html#l00027">gini_gain.hpp:27</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_aaa53d2150636d2e7d4dfde8030e7377d"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#aaa53d2150636d2e7d4dfde8030e7377d">mlpack::tree::DecisionTree::NumericAuxiliarySplitInfo</a></div><div class="ttdeci">NumericSplit::template AuxiliarySplitInfo< ElemType > NumericAuxiliarySplitInfo</div><div class="ttdoc">Note that this class will also hold the members of the NumericSplit and CategoricalSplit AuxiliarySpl...</div><div class="ttdef"><b>Definition:</b> <a href="decision__tree_8hpp_source.html#l00252">decision_tree.hpp:252</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_af9a0efae345916db75d4de55ee800d0d"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#af9a0efae345916db75d4de55ee800d0d">mlpack::tree::DecisionTree::operator=</a></div><div class="ttdeci">DecisionTree & operator=(const DecisionTree &other)</div><div class="ttdoc">Copy another tree. </div></div>
|
|
<div class="ttc" id="gini__gain_8hpp_html"><div class="ttname"><a href="gini__gain_8hpp.html">gini_gain.hpp</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_a7f944c4f36862ab630f8723c5d3f8ee9"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#a7f944c4f36862ab630f8723c5d3f8ee9">mlpack::tree::DecisionTree::~DecisionTree</a></div><div class="ttdeci">~DecisionTree()</div><div class="ttdoc">Clean up memory. </div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_a8f71e2e71f5b875fe1b6b6e3b2286358"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#a8f71e2e71f5b875fe1b6b6e3b2286358">mlpack::tree::DecisionTree::CategoricalAuxiliarySplitInfo</a></div><div class="ttdeci">CategoricalSplit::template AuxiliarySplitInfo< ElemType > CategoricalAuxiliarySplitInfo</div><div class="ttdef"><b>Definition:</b> <a href="decision__tree_8hpp_source.html#l00254">decision_tree.hpp:254</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_a4372a14aa83472be410486c299990816"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#a4372a14aa83472be410486c299990816">mlpack::tree::DecisionTree::Serialize</a></div><div class="ttdeci">void Serialize(Archive &ar, const unsigned int)</div><div class="ttdoc">Serialize the tree. </div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_aa8550c35ef6755b61d2467aa1347f4a6"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#aa8550c35ef6755b61d2467aa1347f4a6">mlpack::tree::DecisionTree::dimensionTypeOrMajorityClass</a></div><div class="ttdeci">size_t dimensionTypeOrMajorityClass</div><div class="ttdoc">The type of the dimension that we have split on (if we are not a leaf). </div><div class="ttdef"><b>Definition:</b> <a href="decision__tree_8hpp_source.html#l00238">decision_tree.hpp:238</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_a8895e26a965b754ac13b1aee3babfcfc"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#a8895e26a965b754ac13b1aee3babfcfc">mlpack::tree::DecisionTree::DecisionTree</a></div><div class="ttdeci">DecisionTree(const MatType &data, const data::DatasetInfo &datasetInfo, const arma::Row< size_t > &labels, const size_t numClasses, const size_t minimumLeafSize=10)</div><div class="ttdoc">Construct the decision tree on the given data and labels, where the data can be both numeric and cate...</div></div>
|
|
<div class="ttc" id="all__categorical__split_8hpp_html"><div class="ttname"><a href="all__categorical__split_8hpp.html">all_categorical_split.hpp</a></div></div>
|
|
<div class="ttc" id="classmlpack_1_1tree_1_1DecisionTree_html_a9a56744b932b6e6c78b4975c9e355896"><div class="ttname"><a href="classmlpack_1_1tree_1_1DecisionTree.html#a9a56744b932b6e6c78b4975c9e355896">mlpack::tree::DecisionTree::Train</a></div><div class="ttdeci">void Train(const MatType &data, const data::DatasetInfo &datasetInfo, const arma::Row< size_t > &labels, const size_t numClasses, const size_t minimumLeafSize=10)</div><div class="ttdoc">Train the decision tree on the given data. </div></div>
|
|
</div><!-- fragment --></div><!-- contents -->
|
|
<!-- start footer part -->
|
|
<hr class="footer"/><address class="footer"><small>
|
|
Generated by  <a href="http://www.doxygen.org/index.html">
|
|
<img class="footer" src="doxygen.png" alt="doxygen"/>
|
|
</a> 1.8.11
|
|
</small></address>
|
|
</body>
|
|
<script type="text/javascript">
|
|
var x = document.getElementsByClassName("formulaDsp");
|
|
var i;
|
|
for (i = 0; i < x.length; i++)
|
|
{
|
|
x[i].width /= 4;
|
|
}
|
|
</script>
|
|
</html>
|