24 lines
567 B
C++
24 lines
567 B
C++
#include "ODTE.h"
|
|
|
|
namespace pywrap {
|
|
ODTE::ODTE() : PyClassifier("odte", "Odte")
|
|
{
|
|
validHyperparameters = { "n_jobs", "n_estimators", "random_state" };
|
|
}
|
|
int ODTE::getNumberOfNodes() const
|
|
{
|
|
return callMethodInt("get_nodes");
|
|
}
|
|
int ODTE::getNumberOfEdges() const
|
|
{
|
|
return callMethodInt("get_leaves");
|
|
}
|
|
int ODTE::getNumberOfStates() const
|
|
{
|
|
return callMethodInt("get_depth");
|
|
}
|
|
std::string ODTE::graph()
|
|
{
|
|
return callMethodString("graph");
|
|
}
|
|
} /* namespace pywrap */ |