Exception if hyperparameters not valid

2023-08-24 12:09:35 +02:00
parent 86ffdfd6f3
commit 1c6af619b5
16 changed files with 47 additions and 13 deletions
--- a/src/BayesNet/Proposal.h
+++ b/src/BayesNet/Proposal.h
@@ -13,6 +13,7 @@ namespace bayesnet {
        Proposal(torch::Tensor& pDataset, vector<string>& features_, string& className_);
        virtual ~Proposal();
    protected:
+        void checkInput(const torch::Tensor& X, const torch::Tensor& y);
        torch::Tensor prepareX(torch::Tensor& X);
        map<string, vector<int>> localDiscretizationProposal(const map<string, vector<int>>& states, Network& model);
        map<string, vector<int>> fit_local_discretization(const torch::Tensor& y);