{
  "_id": "6a2c47593efcd9bda42d9dd0",
  "Package": "tabnet",
  "Title": "Fit 'TabNet' Models for Classification and Regression",
  "Version": "0.9.0",
  "Authors@R": "c(\nperson(\"Daniel\", \"Falbel\", , \"daniel@rstudio.com\", role = \"aut\"),\nperson(, \"RStudio\", role = \"cph\"),\nperson(\"Christophe\", \"Regouby\", , \"christophe.regouby@free.fr\", role = c(\"cre\", \"ctb\")),\nperson(\"Egill\", \"Fridgeirsson\", role = \"ctb\"),\nperson(\"Philipp\", \"Haarmeyer\", role = \"ctb\"),\nperson(\"Sven\", \"Verweij\", role = \"ctb\",\ncomment = c(ORCID = \"0000-0002-5573-3952\"))\n)",
  "Description": "Implements the 'TabNet' model by Sercan O. Arik et al.\n(2019) <doi:10.48550/arXiv.1908.07442> with 'Coherent\nHierarchical Multi-label Classification Networks' by\nGiunchiglia et al.  <doi:10.48550/arXiv.2010.10151> and\nprovides a consistent interface for fitting and creating\npredictions. It's also fully compatible with the 'tidymodels'\necosystem.",
  "License": "MIT + file LICENSE",
  "URL": "https://mlverse.github.io/tabnet/,\nhttps://github.com/mlverse/tabnet",
  "BugReports": "https://github.com/mlverse/tabnet/issues",
  "VignetteBuilder": "knitr",
  "Config/testthat/edition": "3",
  "Config/testthat/parallel": "false",
  "Config/testthat/start-first": "interface, explain, params",
  "Encoding": "UTF-8",
  "Language": "en-US",
  "Config/roxygen2/version": "8.0.0",
  "NeedsCompilation": "no",
  "Packaged": {
    "Date": "2026-06-12 17:46:22 UTC",
    "User": "root"
  },
  "Author": "Daniel Falbel [aut], RStudio [cph], Christophe Regouby [cre,\nctb], Egill Fridgeirsson [ctb], Philipp Haarmeyer [ctb], Sven\nVerweij [ctb] (ORCID: <https://orcid.org/0000-0002-5573-3952>)",
  "Maintainer": "Christophe Regouby <christophe.regouby@free.fr>",
  "Repository": "https://cran.r-universe.dev",
  "Date/Publication": "2026-06-12 16:30:13 UTC",
  "RemoteUrl": "https://github.com/cran/tabnet",
  "RemoteRef": "HEAD",
  "RemoteSha": "6f51ecf80f26a18f9d72b024176df43b3904ab96",
  "MD5sum": "f70ab1b599bf9f17fb89af9c1d5fa3e2",
  "_user": "cran",
  "_type": "src",
  "_file": "tabnet_0.9.0.tar.gz",
  "_fileid": "8c5bde0338bc6187cfa32d62419a694566cd0144f8c4eefb932c3e3f6947f240",
  "_filesize": 6646107,
  "_sha256": "8c5bde0338bc6187cfa32d62419a694566cd0144f8c4eefb932c3e3f6947f240",
  "_created": "2026-06-12T17:46:22.000Z",
  "_published": "2026-06-12T17:52:25.896Z",
  "_distro": "noble",
  "_jobs": [
    {
      "job": 81087498726,
      "time": 233,
      "config": "linux-devel-x86_64",
      "r": "4.7.0",
      "check": "OK",
      "artifact": "7598751925"
    },
    {
      "job": 81087498705,
      "time": 229,
      "config": "linux-release-x86_64",
      "r": "4.6.0",
      "check": "OK",
      "artifact": "7598750780"
    },
    {
      "job": 81086764042,
      "time": 242,
      "config": "source",
      "r": "4.6.0",
      "check": "OK",
      "artifact": "7598670603"
    },
    {
      "job": 81087498661,
      "time": 312,
      "config": "wasm-release",
      "r": "4.6.0",
      "check": "OK",
      "artifact": "7598779434"
    }
  ],
  "_buildurl": "https://github.com/r-universe/cran/actions/runs/27432673322",
  "_status": "success",
  "_host": "GitHub-Actions",
  "_upstream": "https://github.com/cran/tabnet",
  "_commit": {
    "id": "6f51ecf80f26a18f9d72b024176df43b3904ab96",
    "author": "Christophe Regouby <christophe.regouby@free.fr>",
    "committer": "cran-robot <csardi.gabor+cran@gmail.com>",
    "message": "version 0.9.0\n",
    "time": 1781281813
  },
  "_maintainer": {
    "name": "Christophe Regouby",
    "email": "christophe.regouby@free.fr",
    "login": "cregouby",
    "mastodon": "@cregouby@mastodon.roflcopter.fr",
    "description": "",
    "uuid": 10136115
  },
  "_registered": true,
  "_dependencies": [
    {
      "package": "R",
      "version": ">= 3.6",
      "role": "Depends"
    },
    {
      "package": "coro",
      "role": "Imports"
    },
    {
      "package": "data.tree",
      "role": "Imports"
    },
    {
      "package": "dials",
      "role": "Imports"
    },
    {
      "package": "dplyr",
      "role": "Imports"
    },
    {
      "package": "generics",
      "role": "Imports"
    },
    {
      "package": "ggplot2",
      "role": "Imports"
    },
    {
      "package": "hardhat",
      "version": ">= 1.3.0",
      "role": "Imports"
    },
    {
      "package": "magrittr",
      "role": "Imports"
    },
    {
      "package": "methods",
      "role": "Imports"
    },
    {
      "package": "parsnip",
      "role": "Imports"
    },
    {
      "package": "progress",
      "role": "Imports"
    },
    {
      "package": "purrr",
      "role": "Imports"
    },
    {
      "package": "rlang",
      "role": "Imports"
    },
    {
      "package": "stats",
      "role": "Imports"
    },
    {
      "package": "stringr",
      "role": "Imports"
    },
    {
      "package": "tibble",
      "role": "Imports"
    },
    {
      "package": "tidyr",
      "role": "Imports"
    },
    {
      "package": "torch",
      "version": ">= 0.4.0",
      "role": "Imports"
    },
    {
      "package": "tune",
      "role": "Imports"
    },
    {
      "package": "utils",
      "role": "Imports"
    },
    {
      "package": "vctrs",
      "role": "Imports"
    },
    {
      "package": "withr",
      "role": "Imports"
    },
    {
      "package": "zeallot",
      "role": "Imports"
    },
    {
      "package": "cli",
      "role": "Suggests"
    },
    {
      "package": "knitr",
      "role": "Suggests"
    },
    {
      "package": "modeldata",
      "role": "Suggests"
    },
    {
      "package": "patchwork",
      "role": "Suggests"
    },
    {
      "package": "quarto",
      "role": "Suggests"
    },
    {
      "package": "recipes",
      "role": "Suggests"
    },
    {
      "package": "rmarkdown",
      "role": "Suggests"
    },
    {
      "package": "rsample",
      "role": "Suggests"
    },
    {
      "package": "spelling",
      "role": "Suggests"
    },
    {
      "package": "testthat",
      "version": ">= 3.0.0",
      "role": "Suggests"
    },
    {
      "package": "tidymodels",
      "role": "Suggests"
    },
    {
      "package": "tidyverse",
      "role": "Suggests"
    },
    {
      "package": "vip",
      "role": "Suggests"
    },
    {
      "package": "visdat",
      "role": "Suggests"
    },
    {
      "package": "workflows",
      "role": "Suggests"
    },
    {
      "package": "xgboost",
      "role": "Suggests"
    },
    {
      "package": "yardstick",
      "role": "Suggests"
    }
  ],
  "_owner": "cran",
  "_selfowned": false,
  "_usedby": 0,
  "_updates": [
    {
      "week": "2026-05",
      "n": 1
    },
    {
      "week": "2026-24",
      "n": 1
    }
  ],
  "_tags": [
    {
      "name": "0.8.0",
      "date": "2026-01-31"
    },
    {
      "name": "0.9.0",
      "date": "2026-06-12"
    }
  ],
  "_stars": 0,
  "_contributors": [
    {
      "user": "cregouby",
      "count": 6,
      "uuid": 10136115
    }
  ],
  "_userbio": {
    "uuid": 6899542,
    "type": "organization",
    "name": "cran",
    "description": "Unofficial read-only mirror of all CRAN R packages"
  },
  "_downloads": {
    "count": 4558,
    "source": "https://cranlogs.r-pkg.org/downloads/total/last-month/tabnet"
  },
  "_devurl": "https://github.com/mlverse/tabnet",
  "_pkgdown": "https://mlverse.github.io/tabnet/",
  "_searchresults": 96,
  "_rbuild": "4.6.0",
  "_assets": [
    "extra/citation.cff",
    "extra/citation.html",
    "extra/citation.json",
    "extra/citation.txt",
    "extra/contents.json",
    "extra/NEWS.html",
    "extra/NEWS.txt",
    "extra/readme.html",
    "extra/readme.md",
    "extra/tabnet.html",
    "LICENSE",
    "manual.pdf"
  ],
  "_homeurl": "https://github.com/mlverse/tabnet",
  "_realowner": "mlverse",
  "_cranurl": false,
  "_releases": [
    {
      "version": "0.1.0",
      "date": "2021-01-14"
    },
    {
      "version": "0.2.0",
      "date": "2021-06-22"
    },
    {
      "version": "0.3.0",
      "date": "2021-10-11"
    },
    {
      "version": "0.4.0",
      "date": "2023-05-11"
    },
    {
      "version": "0.5.0",
      "date": "2023-12-05"
    },
    {
      "version": "0.6.0",
      "date": "2024-06-15"
    },
    {
      "version": "0.7.0",
      "date": "2025-04-17"
    },
    {
      "version": "0.8.0",
      "date": "2026-01-31"
    },
    {
      "version": "0.9.0",
      "date": "2026-06-12"
    }
  ],
  "_exports": [
    "%>%",
    "attention_width",
    "augment",
    "build_ancestor_matrix_from_outcomes",
    "cat_emb_dim",
    "check_compliant_node",
    "checkpoint_epochs",
    "decision_width",
    "drop_last",
    "encoder_activation",
    "entmax",
    "entmax15",
    "feature_reusage",
    "lr_scheduler",
    "mask_type",
    "mlp_activation",
    "mlp_hidden_multiplier",
    "momentum",
    "nn_aum_loss",
    "nn_mc_loss",
    "nnf_mc_loss",
    "nnf_multilabel_one_hot",
    "node_to_df",
    "num_independent",
    "num_independent_decoder",
    "num_shared",
    "num_shared_decoder",
    "num_steps",
    "optimizer",
    "penalty",
    "sparsemax",
    "sparsemax15",
    "tabnet",
    "tabnet_config",
    "tabnet_explain",
    "tabnet_fit",
    "tabnet_nn",
    "tabnet_pretrain",
    "verbose",
    "virtual_batch_size"
  ],
  "_help": [
    {
      "page": "tabnet_params",
      "title": "Parameters for the tabnet model",
      "topics": [
        "attention_width",
        "decision_width",
        "feature_reusage",
        "mask_type",
        "momentum",
        "num_independent",
        "num_shared",
        "num_steps"
      ]
    },
    {
      "page": "autoplot.tabnet_explain",
      "title": "Plot tabnet_explain mask importance heatmap",
      "topics": [
        "autoplot.tabnet_explain"
      ]
    },
    {
      "page": "autoplot.tabnet_fit",
      "title": "Plot tabnet_fit model loss along epochs",
      "topics": [
        "autoplot.tabnet_fit",
        "autoplot.tabnet_pretrain"
      ]
    },
    {
      "page": "build_ancestor_matrix_from_outcomes",
      "title": "Build ancestor matrix aligned with observed outcome classes",
      "topics": [
        "build_ancestor_matrix_from_outcomes"
      ]
    },
    {
      "page": "tabnet_non_tunable",
      "title": "Non-tunable parameters for the tabnet model",
      "topics": [
        "cat_emb_dim",
        "checkpoint_epochs",
        "drop_last",
        "encoder_activation",
        "lr_scheduler",
        "mlp_activation",
        "mlp_hidden_multiplier",
        "num_independent_decoder",
        "num_shared_decoder",
        "optimizer",
        "penalty",
        "verbose",
        "virtual_batch_size"
      ]
    },
    {
      "page": "check_compliant_node",
      "title": "Check that Node object names are compliant",
      "topics": [
        "check_compliant_node"
      ]
    },
    {
      "page": "entmax15",
      "title": "Alpha-entmax",
      "topics": [
        "entmax",
        "entmax15"
      ]
    },
    {
      "page": "get_constr_output",
      "title": "Apply hierarchy constraints via max-pooling over descendants (MCM)",
      "topics": [
        "get_constr_output"
      ]
    },
    {
      "page": "get_tau",
      "title": "Optimal threshold (tau) computation for 1.5-entmax",
      "topics": [
        "get_tau"
      ]
    },
    {
      "page": "nn_aum_loss",
      "title": "AUM loss",
      "topics": [
        "nn_aum_loss"
      ]
    },
    {
      "page": "nn_mc_loss",
      "title": "Max-Constraint Margin Loss (module)",
      "topics": [
        "nn_mc_loss"
      ]
    },
    {
      "page": "nn_prune_head",
      "title": "Prune top layer(s) of a tabnet network",
      "topics": [
        "nn_prune_head.tabnet_fit",
        "nn_prune_head.tabnet_pretrain"
      ]
    },
    {
      "page": "nnf_mc_loss",
      "title": "Max-Constraint Margin Loss (functional)",
      "topics": [
        "nnf_mc_loss"
      ]
    },
    {
      "page": "nnf_multilabel_one_hot",
      "title": "Convert class_id tensor to binary one-hot tensor",
      "topics": [
        "nnf_multilabel_one_hot"
      ]
    },
    {
      "page": "node_to_df",
      "title": "Turn a Node object into predictor and outcome.",
      "topics": [
        "node_to_df"
      ]
    },
    {
      "page": "predict.tabnet_fit",
      "title": "Predict using 'tabnet'",
      "topics": [
        "augment.tabnet_fit",
        "predict.tabnet_fit"
      ]
    },
    {
      "page": "sparsemax",
      "title": "Sparsemax",
      "topics": [
        "sparsemax",
        "sparsemax15"
      ]
    },
    {
      "page": "tabnet",
      "title": "Parsnip compatible tabnet model",
      "topics": [
        "tabnet"
      ]
    },
    {
      "page": "tabnet_config",
      "title": "Configuration for TabNet models",
      "topics": [
        "tabnet_config"
      ]
    },
    {
      "page": "tabnet_explain",
      "title": "Interpretation metrics from a TabNet model",
      "topics": [
        "tabnet_explain",
        "tabnet_explain.default",
        "tabnet_explain.model_fit",
        "tabnet_explain.tabnet_fit",
        "tabnet_explain.tabnet_pretrain"
      ]
    },
    {
      "page": "tabnet_fit",
      "title": "Tabnet model",
      "topics": [
        "tabnet_fit",
        "tabnet_fit.data.frame",
        "tabnet_fit.default",
        "tabnet_fit.formula",
        "tabnet_fit.Node",
        "tabnet_fit.recipe"
      ]
    },
    {
      "page": "tabnet_nn",
      "title": "TabNet Model Architecture",
      "topics": [
        "tabnet_nn"
      ]
    },
    {
      "page": "tabnet_pretrain",
      "title": "Tabnet model",
      "topics": [
        "tabnet_pretrain",
        "tabnet_pretrain.data.frame",
        "tabnet_pretrain.default",
        "tabnet_pretrain.formula",
        "tabnet_pretrain.Node",
        "tabnet_pretrain.recipe"
      ]
    }
  ],
  "_readme": "https://github.com/cran/tabnet/raw/HEAD/README.md",
  "_rundeps": [
    "base64enc",
    "bit",
    "bit64",
    "bslib",
    "cachem",
    "callr",
    "class",
    "cli",
    "clock",
    "codetools",
    "coro",
    "cpp11",
    "crayon",
    "data.table",
    "data.tree",
    "desc",
    "diagram",
    "dials",
    "DiceDesign",
    "digest",
    "dplyr",
    "evaluate",
    "farver",
    "fastmap",
    "fontawesome",
    "fs",
    "furrr",
    "future",
    "future.apply",
    "GauPro",
    "generics",
    "ggplot2",
    "globals",
    "glue",
    "gower",
    "gtable",
    "hardhat",
    "highr",
    "hms",
    "htmltools",
    "ipred",
    "isoband",
    "jquerylib",
    "jsonlite",
    "KernSmooth",
    "knitr",
    "labeling",
    "lattice",
    "lava",
    "lbfgs",
    "lifecycle",
    "listenv",
    "lubridate",
    "magrittr",
    "MASS",
    "Matrix",
    "memoise",
    "mime",
    "mixopt",
    "modelenv",
    "nnet",
    "numDeriv",
    "otel",
    "parallelly",
    "parsnip",
    "pillar",
    "pkgconfig",
    "prettyunits",
    "processx",
    "prodlim",
    "progress",
    "progressr",
    "ps",
    "purrr",
    "R6",
    "rappdirs",
    "RColorBrewer",
    "Rcpp",
    "RcppArmadillo",
    "recipes",
    "rlang",
    "rmarkdown",
    "rpart",
    "rsample",
    "S7",
    "safetensors",
    "sass",
    "scales",
    "sfd",
    "shape",
    "slider",
    "sparsevctrs",
    "splitfngr",
    "SQUAREM",
    "stringi",
    "stringr",
    "survival",
    "tailor",
    "tibble",
    "tidyr",
    "tidyselect",
    "timechange",
    "timeDate",
    "tinytex",
    "torch",
    "tune",
    "tzdb",
    "utf8",
    "vctrs",
    "viridisLite",
    "warp",
    "withr",
    "workflows",
    "xfun",
    "yaml",
    "yardstick",
    "zeallot"
  ],
  "_vignettes": [
    {
      "source": "tidymodels-interface.Rmd",
      "filename": "tidymodels-interface.html",
      "title": "Fitting tabnet with tidymodels",
      "engine": "knitr::rmarkdown",
      "headings": [],
      "created": "2021-01-14 08:30:02",
      "modified": "2026-01-31 15:40:02",
      "commits": 3
    },
    {
      "source": "Hierarchical_classification.Rmd",
      "filename": "Hierarchical_classification.html",
      "title": "Hierarchical Classification",
      "engine": "knitr::rmarkdown",
      "headings": [
        "Data format",
        "Data preparation",
        "Node preparation rules for {tabnet} models",
        "Avoid factor predictors",
        "Avoid column name collision with reserved {data.tree} names",
        "Avoid column named level_* to avoid collision with output data.tree names",
        "Ensure the last hierarchy of the tree is the observation id",
        "Ensure there is a root level in the hierarchy",
        "Ensure there is no missing values in the hierarchical classes",
        "Model building",
        "Data set split",
        "Model diagnostic",
        "Model inference"
      ],
      "created": "2023-12-06 02:40:19",
      "modified": "2026-06-12 16:30:13",
      "commits": 2
    },
    {
      "source": "interpretation.Rmd",
      "filename": "interpretation.html",
      "title": "Interpretation tools",
      "engine": "knitr::rmarkdown",
      "headings": [
        "Experiments",
        "Datasets",
        "Syn2",
        "Syn 4"
      ],
      "created": "2021-01-14 08:30:02",
      "modified": "2026-01-31 15:40:02",
      "commits": 6
    },
    {
      "source": "selfsupervised_training.Rmd",
      "filename": "selfsupervised_training.html",
      "title": "Self-supervised training and fine-tuning",
      "engine": "knitr::rmarkdown",
      "headings": [
        "Data preprocessing",
        "Self-supervised training step",
        "Continuing training with supervised task",
        "Comparing against a model without pretraining"
      ],
      "created": "2023-12-06 02:40:19",
      "modified": "2026-06-12 16:30:13",
      "commits": 3
    },
    {
      "source": "Missing_data_predictors.Rmd",
      "filename": "Missing_data_predictors.html",
      "title": "Training a Tabnet model from missing-values dataset",
      "author": "Christophe Regouby",
      "engine": "knitr::rmarkdown",
      "headings": [
        "Motivation",
        "Missing-data dataset creation",
        "Ames missings understanding",
        "While keeping some room for freedom",
        "Ames with missing data",
        "Model pretraining",
        "Variable importance with raw ames dataset",
        "Variable importance with ames_missing dataset",
        "Model training",
        "Conclusion"
      ],
      "created": "2023-05-11 07:50:02",
      "modified": "2025-04-17 02:12:05",
      "commits": 3
    },
    {
      "source": "aum_loss.Rmd",
      "filename": "aum_loss.html",
      "title": "Using ROC AUM loss for imbalanced binary classification",
      "engine": "knitr::rmarkdown",
      "headings": [
        "Introduction",
        "How imbalance is my problem ?",
        "Solutions to improve imbalanced classification models",
        "Using the AUC metric and pr_curve() plots",
        "Case-weight",
        "ROC_AUM loss",
        "All together"
      ],
      "created": "2026-01-31 15:40:02",
      "modified": "2026-01-31 15:40:02",
      "commits": 1
    }
  ],
  "_score": 5.4191968042675684,
  "_indexed": false,
  "_nocasepkg": "tabnet",
  "_universes": [
    "cran"
  ],
  "_indexurl": "https://mlverse.r-universe.dev/tabnet",
  "_previous": "0.8.0",
  "_binaries": [
    {
      "r": "4.7.0",
      "os": "linux",
      "version": "0.9.0",
      "date": "2026-06-12T17:49:19.000Z",
      "distro": "noble",
      "commit": "6f51ecf80f26a18f9d72b024176df43b3904ab96",
      "fileid": "36ff625b15734570603137c2fb0a78dd0275c001f6bb10832ebbdcbdab35dc17",
      "status": "success",
      "check": "OK",
      "buildurl": "https://github.com/r-universe/cran/actions/runs/27432673322"
    },
    {
      "r": "4.6.0",
      "os": "linux",
      "version": "0.9.0",
      "date": "2026-06-12T17:49:21.000Z",
      "distro": "noble",
      "commit": "6f51ecf80f26a18f9d72b024176df43b3904ab96",
      "fileid": "6115156ef24a5f7c1c57073c18910481f78aa12600bf413d9f075b6598d21b9d",
      "status": "success",
      "check": "OK",
      "buildurl": "https://github.com/r-universe/cran/actions/runs/27432673322"
    },
    {
      "r": "4.6.0",
      "os": "wasm",
      "version": "0.9.0",
      "date": "2026-06-12T17:52:00.000Z",
      "commit": "6f51ecf80f26a18f9d72b024176df43b3904ab96",
      "fileid": "3312b79c2d777e73d2fb2a8fa517ea9e595e6f604127bd980f8be789630cdd20",
      "status": "success",
      "buildurl": "https://github.com/r-universe/cran/actions/runs/27432673322"
    }
  ]
}