3

mlrパッケージでモデルを実行しようとしていますが、predict()機能に問題があります。次のエラーメッセージが表示されます。

Error in predict(mod, task = task, subset = test) : 
Assertion on 'subset' failed: Must be of type 'integerish', not 'data.frame'

以下の再現可能な例を見つけてください。

require(mlr)     # models
require(caTools) # sampling
require(Zelig)   # data

data("voteincome")
voteincome$vote <- as.factor(voteincome$vote)

set.seed(0)
sample <- sample.split(voteincome, SplitRatio = .75)
train <- subset(voteincome, sample == TRUE)
test <- subset(voteincome, sample == FALSE)

train <- na.omit(train)
test <- na.omit(test)

task <- makeClassifTask(data = train, target = "vote")
lrnr <- makeLearner("classif.randomForest")
mod <- train(lrnr, task)
pred <- predict(mod, task = task, subset = test)

そして、エラーが表示されます。私は何か間違ったことをしていますか?ありがとう!

4

2 に答える 2

1

これを試して:

pred <- predict(mod$learner.model, task = task, subset = test) 
于 2015-12-20T08:31:55.210 に答える