如何打印神經網絡預測概率的混淆矩陣

Question

我需要將概率轉換為真或假（> 0.5），然后打印混淆矩陣。 我找不到如何執行此操作的示例。

在我的嘗試中，我也很難引用轉換后的“成功”目標，現在是“成功真”。

成功是布爾值，Comp 是一個因素（4 個級別），其他是數字。

require(neuralnet)
m <- model.matrix(~Success + Comp + Var2 + Var3 + Var4, data=Train3)
m1 <- model.matrix(~Success + Comp + Var2 + Var3 + Var4, data=Test3)
nn=neuralnet(SuccessTrue~Comp2 + Comp3 + Comp4 + Var2 + Var3 + Var4,data=m, hidden=4, act.fct = "logistic",linear.output = FALSE)
pred <- compute(nn,m1)

Answer 1

我想通了。

> predicted.classes <- ifelse(pred$net.result > 0.5, "TRUE", "FALSE")  
> t <- table(predicted.classes,Test3$Success)  
> confusionMatrix(t)