TabFM 1.0.0 (PyTorch) TabFM is a zero shot tabular foundation model from Google Research. It supports classification and regression on structured/tabular data with mixed numerical and categorical columns, requiring no fine tuning or hyperparameter search training examples are passed as context and predictions are made in a single forward pass. This repository contains the PyTorch weights. For the JAX/Flax weights see google/tabfm 1.0.0 jax. Getting Started Classification: Regression: You can also load directly using the HuggingFace Hub API: Available Checkpoints Subfolder Task is classifier classification/ Classification (up to 10 classes) True regression/ Regression False Developers and Affiliations Developed by the Google Research team. Intended Use Tabular data with numerical and/or categorical columns Binary and multiclass classification (up to 10 classes) Regression on continuous targets Zero shot inference: no dataset specific training or hyperparameter tuning Works with DataFrames (pandas) or numpy arrays Not Intended For Images, audio, video, or raw text More than 10 output classes (hard model limit) Tasks requiring task specific fine tuning Non tabular structured data (gra…
We use cookies for essential functionality and analytics. You can accept or reject analytics cookies.Cookie policy