Add First BayesMetrics Tests

2023-10-05 01:14:16 +02:00
parent 3448fb1299
commit 5f0676691c
4 changed files with 64 additions and 62 deletions
--- a/src/BayesNet/BayesMetrics.cc
+++ b/src/BayesNet/BayesMetrics.cc
@@ -1,7 +1,7 @@
 #include "BayesMetrics.h"
 #include "Mst.h"
 namespace bayesnet {
-    //samples is nxm tensor used to fit the model
+    //samples is n+1xm tensor used to fit the model
    Metrics::Metrics(const torch::Tensor& samples, const vector<string>& features, const string& className, const int classNumStates)
        : samples(samples)
        , features(features)