Remove n_jobs from STree

2023-11-17 10:10:31 +01:00
parent f088df14fd
commit d3cb580387
2 changed files with 3 additions and 4 deletions
--- a/src/PyClassifiers/STree.cc
+++ b/src/PyClassifiers/STree.cc
@@ -8,7 +8,7 @@ namespace pywrap {
    void STree::setHyperparameters(nlohmann::json& hyperparameters)
    {
        // Check if hyperparameters are valid
-        const std::vector<std::string> validKeys = { "C", "n_jobs", "kernel", "max_iter", "max_depth", "random_state", "multiclass_strategy" };
+        const std::vector<std::string> validKeys = { "C", "kernel", "max_iter", "max_depth", "random_state", "multiclass_strategy" };
        checkHyperparameters(validKeys, hyperparameters);
        this->hyperparameters = hyperparameters;
    }