在Y為類別變數的模型做完預測之後(predict),我們會得到的結果不是0或1,而是一種概率,例如需要預測的x1x2x3...輸入完是0的概率為20%,1為70%,2為10%這樣的概率,因此,我們可以簡單的推論為1,這時候在Rcode實現的時候就要用到
max.col(),它可以幫助你判斷20,70,10誰是比較大的數,70是比較大的數所以五們推論為1
當然寫迴圈也做得到,但是R的for跑得慢,是非常不推薦的.
Ypred = predict(result,xdata_test__)
Ypred = t(matrix(Ypred,m,length(Ypred)/m))#轉置
Ypred = levels(fraud_train$fraud_ind)[max.col(Ypred)]
Ypred = factor(Ypred1,levels=levels(fraud_train$fraud_ind))
iThome鐵人賽
看影片追技術