关于weka里面如何获得绘制ROC曲线的数据

在自己课题上常需要使用到weka工具包,以使用randomforest为例说明如何绘制roc曲线。

1)打开weka,选择explorer,open你的arff格式的数据,选择classify,choose选择你使用的分类方法,设置相应的参数,将output设置为plain text, save之后点击start;


2)得到的结果如下图所示,需要单独将prediction on test data这部分数据提取出来进行处理。首先按照prediction值从大到小进行排列,然后提取标签值,可以分别取actual和predicted的第一列或者第二列,处理结果如图所示,然后根据标签计算tpr和fpr,这里标签分别是2和1,对应的样本分别是405和406,因此可以通过python的小脚本获得我们需要的数据。

3)输出结果如下所示,将两列值输入到origin8或者spss里面就可以得到相应roc曲线图及相关的信息。

猜你喜欢

转载自blog.csdn.net/sinat_35853898/article/details/79199896
今日推荐