使用pre()、Rulefit包预测函数使用情况

Question

我尝试使用pre()函数在插入符号中实现规则，以完成分类任务。我使用以下命令训练模型

fit = pre(output~.,data=train)

我假设创建的fit对象是一个模型，该模型具有“规则和原始特征”作为其预测器，并可用于对测试数据进行预测。

我不清楚如何使用这个模型来预测测试数据的输出。

我尝试使用预测命令predict(fit,test_data)对test_data进行预测，结果是get是数值向量，值的范围从.9到1.9

如果有人能在这方面指导我，我会很有帮助。我想知道创建的fit对象是否是一个模型，如果是，如何使用它来预测测试数据

问候你，JJ

Answer 1

如果output是一个二元因子，您应该在调用函数pre()时指定family = binomial。否则，默认情况下将假设响应分布为高斯分布(R中的大多数监督学习方法都是这种情况)。确保在train中将output编码为二进制因子。另请参见?pre。

默认情况下，返回的预测值在线性预测器的范围内，因此它们可以从-Inf到Inf取值。要获得预测概率，您需要为预测类指定type = "response"或type = "class"。另请参见?predict.pre。

可以在此处找到有关软件包pre的简短教程：https://github.com/marjoleinF/pre

Answer 2

来自pre包的参考手册：

使用predict.pre(pre_obj, newdata)