autoplot.tabnet_fit {tabnet} | R Documentation |
Plot tabnet_fit model loss along epochs
Description
Plot tabnet_fit model loss along epochs
Usage
autoplot.tabnet_fit(object, ...)
autoplot.tabnet_pretrain(object, ...)
Arguments
object |
A |
... |
not used. |
Details
Plot the training loss along epochs, and validation loss along epochs if any.
A dot is added on epochs where model snapshot is available, helping
the choice of from_epoch
value for later model training resume.
Value
A ggplot
object.
Examples
## Not run:
library(ggplot2)
data("attrition", package = "modeldata")
attrition_fit <- tabnet_fit(Attrition ~. , data=attrition, valid_split=0.2, epoch=11)
# Plot the model loss over epochs
autoplot(attrition_fit)
## End(Not run)
[Package tabnet version 0.7.0 Index]