Continue refactoring

2025-02-27 09:57:40 +01:00
parent 4e3043b2d1
commit f51d5b5e40
5 changed files with 28 additions and 12 deletions
--- a/src/experimental_clfs/XA1DE.cpp
+++ b/src/experimental_clfs/XA1DE.cpp
@@ -13,7 +13,7 @@ namespace platform {
        auto X = TensorUtils::to_matrix(dataset.slice(0, 0, dataset.size(0) - 1));
        auto y = TensorUtils::to_vector<int>(dataset.index({ -1, "..." }));
        int num_instances = X[0].size();
-        weights_ = torch::full({ m }, 1.0 / m, torch::kFloat64);
+        weights_ = weights;
        normalize_weights(num_instances);
        aode_.fit(X, y, features, className, states, weights_, true);
    }