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

9.7 KiB
Raw Blame History

<html lang="en"> <head> </head>
LCOV - code coverage report
Current view: top level - bayesnet/classifiers - SPODELd.cc (source / functions) Coverage Total Hit
Test: coverage.info Lines: 100.0 % 26 26
Test Date: 2024-04-30 20:26:57 Functions: 100.0 % 6 6 

            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 "SPODELd.h"
       8              : 
       9              : namespace bayesnet {
      10          110 :     SPODELd::SPODELd(int root) : SPODE(root), Proposal(dataset, features, className) {}
      11           84 :     SPODELd& SPODELd::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           84 :         checkInput(X_, y_);
      14           84 :         Xf = X_;
      15           84 :         y = y_;
      16           84 :         return commonFit(features_, className_, states_);
      17              :     }
      18              : 
      19            4 :     SPODELd& SPODELd::fit(torch::Tensor& dataset, const std::vector<std::string>& features_, const std::string& className_, map<std::string, std::vector<int>>& states_)
      20              :     {
      21            4 :         if (!torch::is_floating_point(dataset)) {
      22            2 :             throw std::runtime_error("Dataset must be a floating point tensor");
      23              :         }
      24            8 :         Xf = dataset.index({ torch::indexing::Slice(0, dataset.size(0) - 1), "..." }).clone();
      25            6 :         y = dataset.index({ -1, "..." }).clone().to(torch::kInt32);
      26            2 :         return commonFit(features_, className_, states_);
      27            6 :     }
      28              : 
      29           86 :     SPODELd& SPODELd::commonFit(const std::vector<std::string>& features_, const std::string& className_, map<std::string, std::vector<int>>& states_)
      30              :     {
      31           86 :         features = features_;
      32           86 :         className = className_;
      33              :         // Fills std::vectors Xv & yv with the data from tensors X_ (discretized) & y
      34           86 :         states = fit_local_discretization(y);
      35              :         // We have discretized the input data
      36              :         // 1st we need to fit the model to build the normal SPODE structure, SPODE::fit initializes the base Bayesian network
      37           86 :         SPODE::fit(dataset, features, className, states);
      38           86 :         states = localDiscretizationProposal(states, model);
      39           86 :         return *this;
      40              :     }
      41           68 :     torch::Tensor SPODELd::predict(torch::Tensor& X)
      42              :     {
      43           68 :         auto Xt = prepareX(X);
      44          136 :         return SPODE::predict(Xt);
      45           68 :     }
      46           18 :     std::vector<std::string> SPODELd::graph(const std::string& name) const
      47              :     {
      48           18 :         return SPODE::graph(name);
      49              :     }
      50              : }

Generated by: LCOV version 2.0-1

</html>