时间戳

首页 > 解决方案 > 使用 pre() ，Rulefit 包预测函数使用情况

r - 使用 pre() ，Rulefit 包预测函数使用情况

问题描述

我正在尝试使用该pre()函数进行分类任务的插入符号中的规则拟合实现。我使用以下命令来训练模型

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

我假设创建的拟合对象是一个模型，它具有“规则和原始特征”作为其预测变量，可用于预测测试数据。

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

我尝试使用 predict 命令predict(fit,test_data)进行预测test_data，得到的结果是数字向量，其值范围为 0.9 到 1.9

如果有人可以指导我这将很有帮助。我想创建的拟合对象是否是模型，如果是，如何使用它来预测测试数据

问候，JJ

标签： rpredict

解决方案

如果output是二元因子，则应family = binomial在对 function 的调用中指定pre()。否则，将默认采用高斯响应分布（与 R 中的大多数监督学习方法一样）。确保output被编码为train. 另请参阅?pre。

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

pre可以在这里找到关于包的简短教程： https ://github.com/marjoleinF/pre

推荐阅读