tolerance <- 3

2025-03-08 18:07:56 +01:00
parent 9c1852c6c3
commit ae7b89b134
4 changed files with 19 additions and 7 deletions
--- a/src/experimental_clfs/XA1DE.cpp
+++ b/src/experimental_clfs/XA1DE.cpp
@@ -13,8 +13,8 @@ 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_ = weights;
+        weights_ = torch::full({ num_instances }, 1.0);
        normalize_weights(num_instances);
-        aode_.fit(X, y, features, className, states, weights_, true);
+        aode_.fit(X, y, features, className, states, weights_, true, smoothing);
    }
 }