Refactor ArffFiles library as a git submodule only for tests

2024-05-21 11:50:19 +00:00
parent 22b742f068
commit 0341bd5648
10 changed files with 31 additions and 330 deletions
--- a/bayesnet/classifiers/Proposal.h
+++ b/bayesnet/classifiers/Proposal.h
@@ -27,6 +27,7 @@ namespace bayesnet {
        torch::Tensor y; // y discrete nx1 tensor
        map<std::string, mdlp::CPPFImdlp*> discretizers;
    private:
+        std::vector<int> factorize(const std::vector<std::string>& labels_t);
        torch::Tensor& pDataset; // (n+1)xm tensor
        std::vector<std::string>& pFeatures;
        std::string& pClassName;