12-audioClass.R

# Test set prediction
predXProb <- predict(model, test$X)
predXClass <- speciesClass[apply(predXProb, 1, which.max)]
trueXClass <- speciesClass[apply(test$Y, 1, which.max)]

# Plot confusion matrix
confMatTest <- confusionMatrix(data = factor(predXClass, levels = speciesClass),
                               reference = factor(trueXClass, levels = speciesClass))

pheatmap(confMatTest$table, cluster_rows = F, cluster_cols = F,
         border_color = NA, show_colnames = F,
         labels_row = speciesClass,
         color = cols(max(confMatTest$table)+1))
         
# Accuracy in test set
mean(predXClass == trueXClass) # 0.7364

# Write sessioninfo
writeLines(capture.output(sessionInfo()), "sessionInfo")