{ "_id": "6a6299d9d3420cf5dfdb40ab", "Package": "tabnet", "Title": "Fit 'TabNet' Models for Classification and Regression", "Version": "0.9.1.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 provides\na consistent interface for fitting and creating predictions.\nIt's also fully compatible with the 'tidymodels' ecosystem.", "License": "MIT + file LICENSE", "URL": "https://mlverse.github.io/tabnet/,\nhttps://github.com/mlverse/tabnet", "BugReports": "https://github.com/mlverse/tabnet/issues", "VignetteBuilder": "knitr", "Additional_repositories": "https://bgreenwell.r-universe.dev", "Config/roxygen2/version": "8.0.0", "Config/testthat/edition": "3", "Config/testthat/parallel": "false", "Config/testthat/start-first": "interface, explain, params", "Encoding": "UTF-8", "Language": "en-US", "Roxygen": "list(markdown = TRUE)", "Config/pak/sysreqs": "cmake make libicu-dev libuv1-dev", "Repository": "https://mlverse.r-universe.dev", "Date/Publication": "2026-07-23 21:50:10 UTC", "RemoteUrl": "https://github.com/mlverse/tabnet", "RemoteRef": "HEAD", "RemoteSha": "575daac2baf537516f1fed4a2f18d239576ff02e", "NeedsCompilation": "no", "Packaged": { "Date": "2026-07-23 22:41:56 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 ", "_user": "mlverse", "_type": "src", "_file": "tabnet_0.9.1.9000.tar.gz", "_fileid": "https://r2.ropensci.org/7ff7d7fb0bf99d06e23d22f066ed9b6c3d8d8037aaaf027a5f07adbc2df7d260", "_filesize": 1250017, "_sha256": "7ff7d7fb0bf99d06e23d22f066ed9b6c3d8d8037aaaf027a5f07adbc2df7d260", "_expires": "2026-10-31T22:46:47.000Z", "_created": "2026-07-23T22:41:56.000Z", "_published": "2026-07-23T22:46:49.016Z", "_jobs": [ { "job": 89352256355, "time": 242, "config": "linux-devel-x86_64", "r": "4.7.0", "check": "OK", "artifact": "8581005091" }, { "job": 89352256364, "time": 202, "config": "linux-release-x86_64", "r": "4.6.1", "check": "OK", "artifact": "8580990280" }, { "job": 89352256404, "time": 116, "config": "macos-oldrel-arm64", "r": "4.5.3", "check": "OK", "artifact": "8580961402" }, { "job": 89352256374, "time": 159, "config": "macos-release-arm64", "r": "4.6.1", "check": "OK", "artifact": "8580975416" }, { "job": 89351461956, "time": 277, "config": "source", "r": "4.6.1", "check": "OK", "artifact": "8580919367" }, { "job": 89352256367, "time": 166, "config": "wasm-release", "r": "4.6.0", "check": "OK", "artifact": "8580977225" }, { "job": 89352256388, "time": 151, "config": "windows-devel-x86_64", "r": "4.7.0", "check": "OK", "artifact": "8580973163" }, { "job": 89352256383, "time": 115, "config": "windows-oldrel-x86_64", "r": "4.5.3", "check": "OK", "artifact": "8580960720" }, { "job": 89352256369, "time": 164, "config": "windows-release-x86_64", "r": "4.6.1", "check": "OK", "artifact": "8580977394" } ], "_host": "GitHub-Actions", "_buildurl": "https://github.com/r-universe/mlverse/actions/runs/30050514267", "_status": "success", "_upstream": "https://github.com/mlverse/tabnet", "_commit": { "id": "575daac2baf537516f1fed4a2f18d239576ff02e", "author": "C. Regouby ", "committer": "C. Regouby ", "message": "update version in translation\n", "time": 1784843410 }, "_maintainer": { "name": "Christophe Regouby", "email": "christophe.regouby@free.fr", "login": "cregouby", "mastodon": "@cregouby@mastodon.roflcopter.fr", "description": "", "uuid": 10136115 }, "_distro": "resolute", "_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-35", "n": 1 }, { "week": "2026-05", "n": 3 }, { "week": "2026-23", "n": 1 }, { "week": "2026-24", "n": 7 }, { "week": "2026-30", "n": 3 } ], "_tags": [ { "name": "v0.8.0", "date": "2026-01-31" }, { "name": "v0.9.0", "date": "2026-06-12" }, { "name": "v0.9.1", "date": "2026-07-22" } ], "_topics": [ "tabnet" ], "_stars": 119, "_contributors": [ { "user": "cregouby", "count": 229, "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", "followers": 196, "description": "Open source libraries to scale Data Science" }, "_downloads": { "count": 1724, "source": "https://cranlogs.r-pkg.org/downloads/total/last-month/tabnet" }, "_devurl": "https://github.com/mlverse/tabnet", "_pkgdown": "https://mlverse.github.io/tabnet/", "_searchresults": 92, "_rbuild": "4.6.1", "_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" }, { "version": "0.9.1", "date": "2026-07-22" } ], "_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-07-23 21:46:05", "commits": 5 }, { "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": "2026-07-23 21:46:05", "commits": 9 }, { "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": "2026-07-23 21:46:05", "commits": 5 }, { "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.355053296274406, "_indexed": true, "_nocasepkg": "tabnet", "_universes": [ "mlverse", "cregouby" ], "_binaries": [ { "r": "4.7.0", "os": "linux", "version": "0.9.1.9000", "date": "2026-07-23T22:45:07.000Z", "distro": "resolute", "commit": "575daac2baf537516f1fed4a2f18d239576ff02e", "fileid": "https://r2.ropensci.org/751cf1692411abe4fee0c175911ccffc21abb6283e686ae1d76780dd6d4e79b1", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/30050514267" }, { "r": "4.6.1", "os": "linux", "version": "0.9.1.9000", "date": "2026-07-23T22:44:47.000Z", "distro": "resolute", "commit": "575daac2baf537516f1fed4a2f18d239576ff02e", "fileid": "https://r2.ropensci.org/d6e768e333912bce2573545fee740844fc4c1a2dd14b555f45688a1d75b19966", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/30050514267" }, { "r": "4.5.3", "os": "mac", "version": "0.9.1.9000", "date": "2026-07-23T22:43:35.000Z", "commit": "575daac2baf537516f1fed4a2f18d239576ff02e", "fileid": "https://r2.ropensci.org/3fd474b4680ea7189ccb97f6c2bda4a7876419026f943a84a22baad279caa9e3", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/30050514267" }, { "r": "4.6.1", "os": "mac", "version": "0.9.1.9000", "date": "2026-07-23T22:44:04.000Z", "commit": "575daac2baf537516f1fed4a2f18d239576ff02e", "fileid": "https://r2.ropensci.org/d059c1ce9e78d55277c13fdc005932abd0889cd2a47d8f650c12bafd2a800bf8", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/30050514267" }, { "r": "4.6.0", "os": "wasm", "version": "0.9.1.9000", "date": "2026-07-23T22:45:07.000Z", "commit": "575daac2baf537516f1fed4a2f18d239576ff02e", "fileid": "https://r2.ropensci.org/6851ec5d4095d3d2293b5cd7924e206eecd779e882d9ce6a32781dd7fafb816c", "status": "success", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/30050514267" }, { "r": "4.7.0", "os": "win", "version": "0.9.1.9000", "date": "2026-07-23T22:43:29.000Z", "commit": "575daac2baf537516f1fed4a2f18d239576ff02e", "fileid": "https://r2.ropensci.org/b5f1bd114bfe3a7f49b370432eca2107c43c575f604f1076ab04425bc3b90026", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/30050514267" }, { "r": "4.5.3", "os": "win", "version": "0.9.1.9000", "date": "2026-07-23T22:43:21.000Z", "commit": "575daac2baf537516f1fed4a2f18d239576ff02e", "fileid": "https://r2.ropensci.org/bdb44747a79cca4349a7b3056f72b5e1324b76fa36b0b5867ebf559de4e1782a", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/30050514267" }, { "r": "4.6.1", "os": "win", "version": "0.9.1.9000", "date": "2026-07-23T22:43:34.000Z", "commit": "575daac2baf537516f1fed4a2f18d239576ff02e", "fileid": "https://r2.ropensci.org/9c5eed9a1c572f25d491a513516b5fbbc324329cdcaaaeae02c26de65b7b99f8", "status": "success", "check": "OK", "buildurl": "https://github.com/r-universe/mlverse/actions/runs/30050514267" } ] }