6.5 KiB
6.5 KiB
<html lang="en">
<head>
</head>
</html>
LCOV - code coverage report | ||||||||||||||||||||||
![]() | ||||||||||||||||||||||
|
||||||||||||||||||||||
![]() |
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 "SPODE.h" 8 : 9 : namespace bayesnet { 10 : 11 562 : SPODE::SPODE(int root) : Classifier(Network()), root(root) {} 12 : 13 508 : void SPODE::buildModel(const torch::Tensor& weights) 14 : { 15 : // 0. Add all nodes to the model 16 508 : addNodes(); 17 : // 1. Add edges from the class node to all other nodes 18 : // 2. Add edges from the root node to all other nodes 19 12840 : for (int i = 0; i < static_cast<int>(features.size()); ++i) { 20 12332 : model.addEdge(className, features[i]); 21 12332 : if (i != root) { 22 11824 : model.addEdge(features[root], features[i]); 23 : } 24 : } 25 508 : } 26 34 : std::vector<std::string> SPODE::graph(const std::string& name) const 27 : { 28 34 : return model.graph(name); 29 : } 30 : 31 : } |
![]() |
Generated by: LCOV version 2.0-1 |
</html>