Add getNumberOfNodes & getNumberOfEdges to Models

Add some more tests
This commit is contained in:
2023-07-19 15:05:44 +02:00
parent 1a21015492
commit 2f5bd0ea7e
8 changed files with 66 additions and 6 deletions

View File

@@ -30,6 +30,8 @@ namespace bayesnet {
virtual ~BaseClassifier() = default;
BaseClassifier& fit(vector<vector<int>>& X, vector<int>& y, vector<string>& features, string className, map<string, vector<int>>& states);
void addNodes();
int getNumberOfNodes();
int getNumberOfEdges();
Tensor predict(Tensor& X);
vector<int> predict(vector<vector<int>>& X);
float score(Tensor& X, Tensor& y);