rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 6 Wiki Activity

Return File Library to /lib as it is needed by Local Discretization (factorize)

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-04-30 20:31:14 +02:00
parent 7aeffba740
commit 618a1e539c
148 changed files with 1804 additions and 1769 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
html/bayesnet/classifiers/TAN.cc.gcov.html
View File

@@ -37,7 +37,7 @@
</tr>
<tr>
<td class="headerItem">Test Date:</td>
<td class="headerValue">2024-04-30 13:59:18</td>
<td class="headerValue">2024-04-30 20:26:57</td>
<td></td>
<td class="headerItem">Functions:</td>
<td class="headerCovTableEntryHi">100.0&nbsp;%</td>
@@ -69,40 +69,40 @@
<span id="L7"><span class="lineNum"> 7</span> : #include &quot;TAN.h&quot;</span>
<span id="L8"><span class="lineNum"> 8</span> : </span>
<span id="L9"><span class="lineNum"> 9</span> : namespace bayesnet {</span>
<span id="L10"><span class="lineNum"> 10</span> <span class="tlaGNC tlaBgGNC"> 282 : TAN::TAN() : Classifier(Network()) {}</span></span>
<span id="L10"><span class="lineNum"> 10</span> <span class="tlaGNC tlaBgGNC"> 94 : TAN::TAN() : Classifier(Network()) {}</span></span>
<span id="L11"><span class="lineNum"> 11</span> : </span>
<span id="L12"><span class="lineNum"> 12</span> <span class="tlaGNC"> 78 : void TAN::buildModel(const torch::Tensor&amp; weights)</span></span>
<span id="L12"><span class="lineNum"> 12</span> <span class="tlaGNC"> 26 : void TAN::buildModel(const torch::Tensor&amp; weights)</span></span>
<span id="L13"><span class="lineNum"> 13</span> : {</span>
<span id="L14"><span class="lineNum"> 14</span> : // 0. Add all nodes to the model</span>
<span id="L15"><span class="lineNum"> 15</span> <span class="tlaGNC"> 78 : addNodes();</span></span>
<span id="L15"><span class="lineNum"> 15</span> <span class="tlaGNC"> 26 : addNodes();</span></span>
<span id="L16"><span class="lineNum"> 16</span> : // 1. Compute mutual information between each feature and the class and set the root node</span>
<span id="L17"><span class="lineNum"> 17</span> : // as the highest mutual information with the class</span>
<span id="L18"><span class="lineNum"> 18</span> <span class="tlaGNC"> 78 : auto mi = std::vector &lt;std::pair&lt;int, float &gt;&gt;();</span></span>
<span id="L19"><span class="lineNum"> 19</span> <span class="tlaGNC"> 234 : torch::Tensor class_dataset = dataset.index({ -1, &quot;...&quot; });</span></span>
<span id="L20"><span class="lineNum"> 20</span> <span class="tlaGNC"> 534 : for (int i = 0; i &lt; static_cast&lt;int&gt;(features.size()); ++i) {</span></span>
<span id="L21"><span class="lineNum"> 21</span> <span class="tlaGNC"> 1368 : torch::Tensor feature_dataset = dataset.index({ i, &quot;...&quot; });</span></span>
<span id="L22"><span class="lineNum"> 22</span> <span class="tlaGNC"> 456 : auto mi_value = metrics.mutualInformation(class_dataset, feature_dataset, weights);</span></span>
<span id="L23"><span class="lineNum"> 23</span> <span class="tlaGNC"> 456 : mi.push_back({ i, mi_value });</span></span>
<span id="L24"><span class="lineNum"> 24</span> <span class="tlaGNC"> 456 : }</span></span>
<span id="L25"><span class="lineNum"> 25</span> <span class="tlaGNC"> 1050 : sort(mi.begin(), mi.end(), [](const auto&amp; left, const auto&amp; right) {return left.second &lt; right.second;});</span></span>
<span id="L26"><span class="lineNum"> 26</span> <span class="tlaGNC"> 78 : auto root = mi[mi.size() - 1].first;</span></span>
<span id="L18"><span class="lineNum"> 18</span> <span class="tlaGNC"> 26 : auto mi = std::vector &lt;std::pair&lt;int, float &gt;&gt;();</span></span>
<span id="L19"><span class="lineNum"> 19</span> <span class="tlaGNC"> 78 : torch::Tensor class_dataset = dataset.index({ -1, &quot;...&quot; });</span></span>
<span id="L20"><span class="lineNum"> 20</span> <span class="tlaGNC"> 178 : for (int i = 0; i &lt; static_cast&lt;int&gt;(features.size()); ++i) {</span></span>
<span id="L21"><span class="lineNum"> 21</span> <span class="tlaGNC"> 456 : torch::Tensor feature_dataset = dataset.index({ i, &quot;...&quot; });</span></span>
<span id="L22"><span class="lineNum"> 22</span> <span class="tlaGNC"> 152 : auto mi_value = metrics.mutualInformation(class_dataset, feature_dataset, weights);</span></span>
<span id="L23"><span class="lineNum"> 23</span> <span class="tlaGNC"> 152 : mi.push_back({ i, mi_value });</span></span>
<span id="L24"><span class="lineNum"> 24</span> <span class="tlaGNC"> 152 : }</span></span>
<span id="L25"><span class="lineNum"> 25</span> <span class="tlaGNC"> 350 : sort(mi.begin(), mi.end(), [](const auto&amp; left, const auto&amp; right) {return left.second &lt; right.second;});</span></span>
<span id="L26"><span class="lineNum"> 26</span> <span class="tlaGNC"> 26 : auto root = mi[mi.size() - 1].first;</span></span>
<span id="L27"><span class="lineNum"> 27</span> : // 2. Compute mutual information between each feature and the class</span>
<span id="L28"><span class="lineNum"> 28</span> <span class="tlaGNC"> 78 : auto weights_matrix = metrics.conditionalEdge(weights);</span></span>
<span id="L28"><span class="lineNum"> 28</span> <span class="tlaGNC"> 26 : auto weights_matrix = metrics.conditionalEdge(weights);</span></span>
<span id="L29"><span class="lineNum"> 29</span> : // 3. Compute the maximum spanning tree</span>
<span id="L30"><span class="lineNum"> 30</span> <span class="tlaGNC"> 78 : auto mst = metrics.maximumSpanningTree(features, weights_matrix, root);</span></span>
<span id="L30"><span class="lineNum"> 30</span> <span class="tlaGNC"> 26 : auto mst = metrics.maximumSpanningTree(features, weights_matrix, root);</span></span>
<span id="L31"><span class="lineNum"> 31</span> : // 4. Add edges from the maximum spanning tree to the model</span>
<span id="L32"><span class="lineNum"> 32</span> <span class="tlaGNC"> 456 : for (auto i = 0; i &lt; mst.size(); ++i) {</span></span>
<span id="L33"><span class="lineNum"> 33</span> <span class="tlaGNC"> 378 : auto [from, to] = mst[i];</span></span>
<span id="L34"><span class="lineNum"> 34</span> <span class="tlaGNC"> 378 : model.addEdge(features[from], features[to]);</span></span>
<span id="L32"><span class="lineNum"> 32</span> <span class="tlaGNC"> 152 : for (auto i = 0; i &lt; mst.size(); ++i) {</span></span>
<span id="L33"><span class="lineNum"> 33</span> <span class="tlaGNC"> 126 : auto [from, to] = mst[i];</span></span>
<span id="L34"><span class="lineNum"> 34</span> <span class="tlaGNC"> 126 : model.addEdge(features[from], features[to]);</span></span>
<span id="L35"><span class="lineNum"> 35</span> : }</span>
<span id="L36"><span class="lineNum"> 36</span> : // 5. Add edges from the class to all features</span>
<span id="L37"><span class="lineNum"> 37</span> <span class="tlaGNC"> 534 : for (auto feature : features) {</span></span>
<span id="L38"><span class="lineNum"> 38</span> <span class="tlaGNC"> 456 : model.addEdge(className, feature);</span></span>
<span id="L39"><span class="lineNum"> 39</span> <span class="tlaGNC"> 456 : }</span></span>
<span id="L40"><span class="lineNum"> 40</span> <span class="tlaGNC"> 612 : }</span></span>
<span id="L41"><span class="lineNum"> 41</span> <span class="tlaGNC"> 12 : std::vector&lt;std::string&gt; TAN::graph(const std::string&amp; title) const</span></span>
<span id="L37"><span class="lineNum"> 37</span> <span class="tlaGNC"> 178 : for (auto feature : features) {</span></span>
<span id="L38"><span class="lineNum"> 38</span> <span class="tlaGNC"> 152 : model.addEdge(className, feature);</span></span>
<span id="L39"><span class="lineNum"> 39</span> <span class="tlaGNC"> 152 : }</span></span>
<span id="L40"><span class="lineNum"> 40</span> <span class="tlaGNC"> 204 : }</span></span>
<span id="L41"><span class="lineNum"> 41</span> <span class="tlaGNC"> 4 : std::vector&lt;std::string&gt; TAN::graph(const std::string&amp; title) const</span></span>
<span id="L42"><span class="lineNum"> 42</span> : {</span>
<span id="L43"><span class="lineNum"> 43</span> <span class="tlaGNC"> 12 : return model.graph(title);</span></span>
<span id="L43"><span class="lineNum"> 43</span> <span class="tlaGNC"> 4 : return model.graph(title);</span></span>
<span id="L44"><span class="lineNum"> 44</span> : }</span>
<span id="L45"><span class="lineNum"> 45</span> : }</span>
</pre>