From b055065e5996eb9b27241773a30e48f30efa8b7a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?= Date: Wed, 26 Feb 2025 21:08:33 +0100 Subject: [PATCH] Fix predict_proba declaration --- src/experimental_clfs/ExpClf.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/experimental_clfs/ExpClf.h b/src/experimental_clfs/ExpClf.h index f9e5cb9..65d3172 100644 --- a/src/experimental_clfs/ExpClf.h +++ b/src/experimental_clfs/ExpClf.h @@ -32,7 +32,7 @@ namespace platform { torch::Tensor predict(torch::Tensor& X) override; torch::Tensor predict_proba(torch::Tensor& X) override; std::vector predict_spode(std::vector>& test_data, int parent); - std::vector> predict_proba(std::vector>& X) override; + std::vector> predict_proba(const std::vector>& X); float score(std::vector>& X, std::vector& y) override; float score(torch::Tensor& X, torch::Tensor& y) override; int getNumberOfNodes() const override; @@ -47,7 +47,7 @@ namespace platform { std::vector getNotes() const override { return notes; } std::vector graph(const std::string& title = "") const override { return {}; } void setHyperparameters(const nlohmann::json& hyperparameters) override; - void set_active_parents(std::vector active_parents) { for (const auto& parent : active_parents) aode_.add_active_parent(parent); } + void set_active_parents(const std::vector active_parents) { for (const auto& parent : active_parents) aode_.add_active_parent(parent); } void add_active_parent(int parent) { aode_.add_active_parent(parent); } void remove_last_parent() { aode_.remove_last_parent(); } protected: