Refactor Hyperparameters management
This commit is contained in:
@@ -1,15 +1,12 @@
|
||||
#include "ODTE.h"
|
||||
|
||||
namespace pywrap {
|
||||
ODTE::ODTE() : PyClassifier("odte", "Odte")
|
||||
{
|
||||
validHyperparameters = { "n_jobs", "n_estimators", "random_state" };
|
||||
}
|
||||
std::string ODTE::graph()
|
||||
{
|
||||
return callMethodString("graph");
|
||||
}
|
||||
void ODTE::setHyperparameters(const nlohmann::json& hyperparameters)
|
||||
{
|
||||
// Check if hyperparameters are valid
|
||||
const std::vector<std::string> validKeys = { "n_jobs", "n_estimators", "random_state" };
|
||||
checkHyperparameters(validKeys, hyperparameters);
|
||||
this->hyperparameters = hyperparameters;
|
||||
}
|
||||
} /* namespace pywrap */
|
Reference in New Issue
Block a user