ограничить параметры состояния CUBIST

Question

Существует алгоритм CUBIST для генерации моделей на основе правил.

он доступен в качестве источника C: https://www.rulequest.com/cubist-win.html

и R пакет: http://cran.r-project.org/web/packages/Cubist/index.html

он создает правила, которые выглядят следующим образом:

if 
    class in {CAR, VAN} 
    displ <= 4.6 
    fuel in {R, D, C} 
then 
    fuel cost = -49.2 + 163 cylinders + 1306 displ/cyl + 77 displ 
       + 46 valves/cyl 

Теперь то, что мне нужно, чтобы определить, какие входных параметров можно использовать в состоянии и которые могут быть использованы в формуле. Исходный код не позволяет этого и разбивает параметры самостоятельно.

Я посмотрел код C, и это довольно сложно.

Есть ли более простой способ решить мою проблему (помимо изменения кода C)? Я попытался найти такие модификации в сети, но не смог.

Answer 1

Собственно, параметры могут быть настроены. Вот пример от https://cran.r-project.org/web/packages/Cubist/vignettes/cubist.pdf

library(caret) 
    set.seed(1) 
    cTune <- train(x = trainingPredictors, y = trainingOutcome, 
        "cubist", 
        tuneGrid = expand.grid(.committees = c(1, 10, 50, 100), 
              .neighbors = c(0, 1, 5, 9)), 
        trControl = trainControl(method = "cv")) 
    cTune