{ "_id": "6a2c45e03efcd9bda42d9266", "Package": "tabnet", "Title": "Fit 'TabNet' Models for Classification and Regression", "Version": "0.9.0.9000", "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) with 'Coherent\nHierarchical Multi-label Classification Networks' by\nGiunchiglia et al. 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", "Roxygen": "list(markdown = TRUE)", "Language": "en-US", "Config/roxygen2/version": "8.0.0", "Config/pak/sysreqs": "cmake make libicu-dev libuv1-dev", "Repository": "https://mlverse.r-universe.dev", "Date/Publication": "2026-06-12 16:24:02 UTC", "RemoteUrl": "https://github.com/mlverse/tabnet", "RemoteRef": "HEAD", "RemoteSha": "71c4ca1d737cc7d6b1d8e521be23b23050ee1b42", "NeedsCompilation": "no", "Packaged": { "Date": "2026-06-12 17:41:18 UTC", "User": "root" }, "Author": "Daniel Falbel [aut],\nRStudio [cph],\nChristophe Regouby [cre, ctb],\nEgill Fridgeirsson [ctb],\nPhilipp Haarmeyer [ctb],\nSven Verweij [ctb] (ORCID: )", "Maintainer": "Christophe Regouby ", "MD5sum": "bb4b66b3a681480ceb8fc32c64e468ef", "_user": "mlverse", "_type": "src", "_file": "tabnet_0.9.0.9000.tar.gz", "_fileid": "89afee18303e10f9a9eb5a41255134c0300894acd2370c9984d522177e1cb9f1", "_filesize": 1250544, "_sha256": "89afee18303e10f9a9eb5a41255134c0300894acd2370c9984d522177e1cb9f1", "_created": "2026-06-12T17:41:18.000Z", "_published": "2026-06-12T17:46:08.236Z", "_distro": "noble", "_jobs": [ { "job": 81086587768, "time": 244, "config": "linux-devel-x86_64", "r": "4.7.0", "check": "OK", "artifact": "7598649280" }, { "job": 81086587782, "time": 243, "config": "linux-release-x86_64", "r": "4.6.0", "check": "OK", "artifact": "7598650114" }, { "job": 81086587797, "time": 202, "config": "macos-oldrel-arm64", "r": "4.5.3", "check": "OK", "artifact": "7598635079" }, { "job": 81086587777, "time": 155, "config": "macos-release-arm64", "r": "4.6.0", "check": "OK", "artifact": "7598619183" }, { "job": 81085815234, "time": 265, "config": "source", "r": "4.6.0", "check": "OK", "artifact": "7598565859" }, { "job": 81086587748, "time": 162, "config": "wasm-release", "r": "4.6.0", "check": "OK", "artifact": "7598621262" }, { "job": 81086587769, "time": 169, "config": "windows-devel", "r": "4.7.0", "check": "OK", "artifact": "7598624094" }, { "job": 81086587755, "time": 172, "config": "windows-oldrel", "r": "4.5.3", "check": "OK", "artifact": "7598625343" }, { "job": 81086587765, "time": 157, "config": "windows-release", "r": "4.6.0", "check": "OK", "artifact": "7598620353" } ], "_buildurl": "https://github.com/r-universe/mlverse/actions/runs/27432395964", "_status": "success", "_host": "GitHub-Actions", "_upstream": "https://github.com/mlverse/tabnet", "_commit": { "id": "71c4ca1d737cc7d6b1d8e521be23b23050ee1b42", "author": "C. Regouby ", "committer": "C. Regouby ", "message": "Increment version number to 0.9.0.9000\n", "time": 1781281442 }, "_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": "mlverse", "_selfowned": true, "_usedby": 0, "_updates": [ { "week": "2025-24", "n": 1 }, { "week": "2025-25", "n": 5 }, { "week": "2025-27", "n": 1 }, { "week": "2025-35", "n": 1 }, { "week": "2026-05", "n": 3 }, { "week": "2026-23", "n": 1 }, { "week": "2026-24", "n": 7 } ], "_tags": [ { "name": "v0.8.0", "date": "2026-01-31" }, { "name": "v0.9.0", "date": "2026-06-12" } ], "_topics": [ "tabnet" ], "_stars": 116, "_contributors": [ { "user": "cregouby", "count": 226, "uuid": 10136115 }, { "user": "dfalbel", "count": 159, "uuid": 4706822 }, { "user": "christophe-regouby", "count": 87, "uuid": 167972538 }, { "user": "cmcmaster1", "count": 18, "uuid": 30722043 }, { "user": "egillax", "count": 4, "uuid": 24678081 }, { "user": "sebffischer", "count": 3, "uuid": 20975557 }, { "user": "svenvw", "count": 1, "uuid": 37927107 }, { "user": "cphaarmeyer", "count": 1, "uuid": 49557518 }, { "user": "zerweck", "count": 1, "uuid": 5678151 } ], "_userbio": { "uuid": 55406849, "type": "organization", "name": "mlverse", "description": "Open source libraries to scale Data Science" }, "_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": true, "_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/mlverse/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": "2020-12-18 12:02:30", "modified": "2025-04-25 15:08:54", "commits": 4 }, { "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-07-22 10:48:16", "modified": "2026-06-12 13:39:46", "commits": 4 }, { "source": "interpretation.Rmd", "filename": "interpretation.html", "title": "Interpretation tools", "engine": "knitr::rmarkdown", "headings": [ "Experiments", "Datasets", "Syn2", "Syn 4" ], "created": "2021-01-07 18:43:59", "modified": "2025-08-25 07:34:29", "commits": 8 }, { "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-07-22 10:48:16", "modified": "2026-06-12 13:39:46", "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": "2022-02-11 17:44:42", "modified": "2025-03-30 20:53:21", "commits": 4 }, { "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": "2025-08-25 07:34:29", "modified": "2025-08-25 07:34:29", "commits": 1 } ], "_score": 9.930812824836707, "_indexed": true, "_nocasepkg": "tabnet", "_universes": [ "mlverse", "cregouby" ], "_previous": "0.8.0.9000", "_binaries": [ { "r": "4.7.0", "os": "linux", "version": "0.9.0.9000", "date": "2026-06-12T17:44:26.000Z", "distro": "noble", "commit": "71c4ca1d737cc7d6b1d8e521be23b23050ee1b42", "fileid": "2b3d143b5e03102186d12ceac82c4bf5189cc4ea5c16800b27ead80847b1b3f5", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/27432395964" }, { "r": "4.6.0", "os": "linux", "version": "0.9.0.9000", "date": "2026-06-12T17:44:29.000Z", "distro": "noble", "commit": "71c4ca1d737cc7d6b1d8e521be23b23050ee1b42", "fileid": "8c7dd4f872b3a10238f10a08b6e39c0fa77e7b29e095cac5db9b1a4b0c09f5d9", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/27432395964" }, { "r": "4.5.3", "os": "mac", "version": "0.9.0.9000", "date": "2026-06-12T17:43:48.000Z", "commit": "71c4ca1d737cc7d6b1d8e521be23b23050ee1b42", "fileid": "65ab71f95e739278f086a2da34a36bd258849352d7e3805be5d3246ba83ed03d", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/27432395964" }, { "r": "4.6.0", "os": "mac", "version": "0.9.0.9000", "date": "2026-06-12T17:43:18.000Z", "commit": "71c4ca1d737cc7d6b1d8e521be23b23050ee1b42", "fileid": "bf2b3cb003f5398280b8ae19f0c78d48b25878a01ef76716c6c8d7ee7168d0e3", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/27432395964" }, { "r": "4.6.0", "os": "wasm", "version": "0.9.0.9000", "date": "2026-06-12T17:44:23.000Z", "commit": "71c4ca1d737cc7d6b1d8e521be23b23050ee1b42", "fileid": "6eebb1b8d29fbbe2c56755e0608ed25d748d4ffde621020c2ac186fbb45426fd", "status": "success", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/27432395964" }, { "r": "4.7.0", "os": "win", "version": "0.9.0.9000", "date": "2026-06-12T17:43:04.000Z", "commit": "71c4ca1d737cc7d6b1d8e521be23b23050ee1b42", "fileid": "c5cdaeb741d9833af2e8a96a8cdd65d26d718be496e14f6852cb5858ad184f11", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/27432395964" }, { "r": "4.5.3", "os": "win", "version": "0.9.0.9000", "date": "2026-06-12T17:42:57.000Z", "commit": "71c4ca1d737cc7d6b1d8e521be23b23050ee1b42", "fileid": "dc7fcb4b717b87c483286efb90f2461cbe7013238f6d65c1a2bca726d78fd6b1", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/27432395964" }, { "r": "4.6.0", "os": "win", "version": "0.9.0.9000", "date": "2026-06-12T17:42:48.000Z", "commit": "71c4ca1d737cc7d6b1d8e521be23b23050ee1b42", "fileid": "83e301728e74c8b98a11c06dd342f3486918fc4625ca09843c820f9687c2c9e8", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/27432395964" } ] }