# 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") |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment