11-audioClass.R

# Grep species, set colors for heatmap
speciesClass <- gsub(colnames(train$Y), pat = "species", rep = "")
cols <- colorRampPalette(rev(brewer.pal(n = 7, name = "RdGy")))
        
# Validation predictions
predProb <- predict(model, val$X)
predClass <- speciesClass[apply(predProb, 1, which.max)]
trueClass <- speciesClass[apply(val$Y, 1, which.max)]

# Plot confusion matrix
confMat <- confusionMatrix(data = factor(predClass, levels = speciesClass),
                           reference = factor(trueClass, levels = speciesClass))

pheatmap(confMat$table, cluster_rows = F, cluster_cols = F,
         border_color = NA, show_colnames = F,
         labels_row = speciesClass,
         color = cols(max(confMat$table)+1))

# Accuracy in validation set
mean(predClass == trueClass) # 0.7541