autoplot.tabnet_fit: Plot tabnet_fit model loss along epochs

In tabnet: Fit 'TabNet' Models for Classification and Regression

Plot tabnet_fit model loss along epochs

Description

Plot tabnet_fit model loss along epochs

Usage

## S3 method for class 'tabnet_fit'
autoplot(object, ...)

## S3 method for class 'tabnet_pretrain'
autoplot(object, ...)

Arguments

object	A tabnet_fit or tabnet_pretrain object as a result of tabnet_fit() or tabnet_pretrain().
...	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

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)

tabnet documentation built on June 22, 2024, 10:22 a.m.