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.h
+++ b/bayesnet/ensembles/BoostAODE.h
@@ -16,7 +16,6 @@ namespace bayesnet {
        void trainModel(const torch::Tensor& weights) override;
    private:
        std::vector<int> initializeModels();
-        torch::Tensor dataset_; // Backup the original dataset
        torch::Tensor X_train, y_train, X_test, y_test;
        // Hyperparameters
        bool bisection = false; // if true, use bisection stratety to add k models at once to the ensemble