3

ROCR を使用して、データセットに対するさまざまな分類アルゴリズムのパフォーマンスを測定しています。次のコード サンプルを使用すると、AUC メトリックを比較的簡単に取得できます。

predictions <- predict(rfmodel, test, type="prob")
pred.obj <- prediction(predictions[,2], data$response)
rfperf <- performance(pred.obj, "tpr","fpr")
print(sprintf("random forest AUC %f", as.numeric(performance(pred.obj,"auc")@y.values)))

論文の表に含めることができる単一の値として、精度、再現率、f などの他の測定値を取得するのに苦労しています。私は次のことを試しました：

> p <- performance(pred.obj,"prec", "rec")
> as.numeric(p@x.values)
Error: (list) object cannot be coerced to type 'double'

値をプロットできますが、これは私が望むものではありません。何か案は？

4

2 に答える 2

3

問題は、出力がリストであることです。as.numeric( unlist ( p@x.values) ) を試してください

于 2013-02-12T09:57:50.123 に答える
2

これは、関数 perf の戻りオブジェクトの構造に関係しています。str を使用すると役立ちます。

p@x.values は、数値のベクトルである 1 つの要素を含むリストです。使うだけ

p@x.values[[1]]

ベクトルを抽出します。

于 2013-02-12T09:55:10.930 に答える