Optimize ComputeCPT method with a approx. 30% reducing time

This commit is contained in:
2025-05-19 17:00:07 +02:00
parent 250036f224
commit 36ce6effe9
6 changed files with 29 additions and 28 deletions

View File

@@ -23,6 +23,7 @@ namespace bayesnet {
throw std::invalid_argument("y must be an integer tensor");
}
}
// Fit method for single classifier
map<std::string, std::vector<int>> Proposal::localDiscretizationProposal(const map<std::string, std::vector<int>>& oldStates, Network& model)
{
// order of local discretization is important. no good 0, 1, 2...

View File

@@ -45,7 +45,6 @@ namespace bayesnet {
}
torch::Tensor SPODELd::predict_proba(torch::Tensor& X)
{
std::cout << "Debug: SPODELd::predict_proba" << std::endl;
auto Xt = prepareX(X);
return SPODE::predict_proba(Xt);
}