Fix metrics error in BoostAODE Convergence

Update algorithm
This commit is contained in:
2024-03-20 23:33:02 +01:00
parent 5826702fc7
commit 6e854dfda3
6 changed files with 98 additions and 65 deletions

View File

@@ -24,7 +24,7 @@ namespace bayesnet {
std::vector<int> Metrics::SelectKBestWeighted(const torch::Tensor& weights, bool ascending, unsigned k)
{
// Return the K Best features
auto n = samples.size(0) - 1;
auto n = features.size();
if (k == 0) {
k = n;
}