Remove dataset clone in BoostAODE

2024-03-21 19:35:08 +01:00
parent 58d5a35a35
commit bc0b938cfc
2 changed files with 0 additions and 3 deletions
--- a/bayesnet/ensembles/BoostAODE.cc
+++ b/bayesnet/ensembles/BoostAODE.cc
@@ -40,8 +40,6 @@ namespace bayesnet {
        if (convergence) {
            // Prepare train & validation sets from train data
            auto fold = folding::StratifiedKFold(5, y_, 271);
-            // save input dataset
-            dataset_ = torch::clone(dataset);
            auto [train, test] = fold.getFold(0);
            auto train_t = torch::tensor(train);
            auto test_t = torch::tensor(test);