rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 6 Wiki Activity
Files
3c7382a93ae5eb9fc06d40534ee2f6a5b7477fa7
BayesNet/html/bayesnet/classifiers/TANLd.cc.gcov.html

7.5 KiB
Raw Blame History

<html lang="en"> <head> </head>
LCOV - code coverage report
Current view: top level - bayesnet/classifiers - TANLd.cc (source / functions) Coverage Total Hit
Test: coverage.info Lines: 100.0 % 17 17
Test Date: 2024-04-30 13:59:18 Functions: 100.0 % 4 4 

            Line data    Source code
       1              : // ***************************************************************
       2              : // SPDX-FileCopyrightText: Copyright 2024 Ricardo Montañana Gómez
       3              : // SPDX-FileType: SOURCE
       4              : // SPDX-License-Identifier: MIT
       5              : // ***************************************************************
       6              : 
       7              : #include "TANLd.h"
       8              : 
       9              : namespace bayesnet {
      10          102 :     TANLd::TANLd() : TAN(), Proposal(dataset, features, className) {}
      11           30 :     TANLd& TANLd::fit(torch::Tensor& X_, torch::Tensor& y_, const std::vector<std::string>& features_, const std::string& className_, map<std::string, std::vector<int>>& states_)
      12              :     {
      13           30 :         checkInput(X_, y_);
      14           30 :         features = features_;
      15           30 :         className = className_;
      16           30 :         Xf = X_;
      17           30 :         y = y_;
      18              :         // Fills std::vectors Xv & yv with the data from tensors X_ (discretized) & y
      19           30 :         states = fit_local_discretization(y);
      20              :         // We have discretized the input data
      21              :         // 1st we need to fit the model to build the normal TAN structure, TAN::fit initializes the base Bayesian network
      22           30 :         TAN::fit(dataset, features, className, states);
      23           30 :         states = localDiscretizationProposal(states, model);
      24           30 :         return *this;
      25              : 
      26              :     }
      27           24 :     torch::Tensor TANLd::predict(torch::Tensor& X)
      28              :     {
      29           24 :         auto Xt = prepareX(X);
      30           48 :         return TAN::predict(Xt);
      31           24 :     }
      32            6 :     std::vector<std::string> TANLd::graph(const std::string& name) const
      33              :     {
      34            6 :         return TAN::graph(name);
      35              :     }
      36              : }

Generated by: LCOV version 2.0-1

</html>