Fix adult numeric features mistake

2025-08-21 19:01:10 +02:00
parent f1a2349245
commit 1aa3b609e5
3 changed files with 14 additions and 1 deletions
--- a/tests/TestUtils.cc
+++ b/tests/TestUtils.cc
@@ -213,6 +213,8 @@ void RawDatasets::loadDataset(const std::string& name, bool class_last)
            if (!is_numeric.at(i)) {
                states[features[i]] = std::vector<int>(maxValues[features[i]]);
                iota(begin(states.at(features[i])), end(states.at(features[i])), 0);
+            } else {
+                states[features[i]] = std::vector<int>();
            }
        }
        yt = torch::tensor(yv, torch::kInt32);