autogluon / tabpfn-mix-1.0-regressor

huggingface.co
Total runs: 3.7K
24-hour runs: 0
7-day runs: -4.7K
30-day runs: -4.9K
Model's Last Updated: 2024年11月27日
tabular-regression

Introduction of tabpfn-mix-1.0-regressor

Model Details of tabpfn-mix-1.0-regressor

TabPFNMix Regressor

TabPFNMix regressor is a tabular foundation model that is pre-trained on purely synthetic datasets sampled from a mix of random regressors.

Architecture

TabPFNMix is based on a 12-layer encoder-decoder Transformer of 37 M parameters. We use a pre-training strategy incorporating in-context learning, similar to that used by TabPFN and TabForestPFN.

Usage

To use TabPFNMix regressor, install AutoGluon by running:

pip install autogluon

A minimal example showing how to perform fine-tuning and inference using TabPFNMix regressor

import pandas as pd

from autogluon.tabular import TabularPredictor


if __name__ == '__main__':  
    train_data = pd.read_csv('https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv')
    subsample_size = 5000
    if subsample_size is not None and subsample_size < len(train_data):
        train_data = train_data.sample(n=subsample_size, random_state=0)
    test_data = pd.read_csv('https://autogluon.s3.amazonaws.com/datasets/Inc/test.csv')

    tabpfnmix_default = {
        "model_path_classifier": "autogluon/tabpfn-mix-1.0-classifier",
        "model_path_regressor": "autogluon/tabpfn-mix-1.0-regressor",
        "n_ensembles": 1,
        "max_epochs": 30,
    }

    hyperparameters = {
        "TABPFNMIX": [
            tabpfnmix_default,
        ],
    }

    label = "age"
    problem_type = "regression"

    predictor = TabularPredictor(
        label=label,
        problem_type=problem_type,
    )
    predictor = predictor.fit(
        train_data=train_data,
        hyperparameters=hyperparameters,
        verbosity=3,
    )

    predictor.leaderboard(test_data, display=True)
Citation

If you find TabPFNMix useful for your research, please consider citing the associated papers:

@article{erickson2020autogluon,
  title={Autogluon-tabular: Robust and accurate automl for structured data},
  author={Erickson, Nick and Mueller, Jonas and Shirkov, Alexander and Zhang, Hang and Larroy, Pedro and Li, Mu and Smola, Alexander},
  journal={arXiv preprint arXiv:2003.06505},
  year={2020}
}

@article{hollmann2022tabpfn,
  title={Tabpfn: A transformer that solves small tabular classification problems in a second},
  author={Hollmann, Noah and M{\"u}ller, Samuel and Eggensperger, Katharina and Hutter, Frank},
  journal={arXiv preprint arXiv:2207.01848},
  year={2022}
}

@article{breejen2024context,
  title={Why In-Context Learning Transformers are Tabular Data Classifiers},
  author={Breejen, Felix den and Bae, Sangmin and Cha, Stephen and Yun, Se-Young},
  journal={arXiv preprint arXiv:2405.13396},
  year={2024}
}
License

This project is licensed under the Apache-2.0 License.

Runs of autogluon tabpfn-mix-1.0-regressor on huggingface.co

3.7K
Total runs
0
24-hour runs
-1.5K
3-day runs
-4.7K
7-day runs
-4.9K
30-day runs

More Information About tabpfn-mix-1.0-regressor huggingface.co Model

More tabpfn-mix-1.0-regressor license Visit here:

https://choosealicense.com/licenses/apache-2.0

tabpfn-mix-1.0-regressor huggingface.co

tabpfn-mix-1.0-regressor huggingface.co is an AI model on huggingface.co that provides tabpfn-mix-1.0-regressor's model effect (), which can be used instantly with this autogluon tabpfn-mix-1.0-regressor model. huggingface.co supports a free trial of the tabpfn-mix-1.0-regressor model, and also provides paid use of the tabpfn-mix-1.0-regressor. Support call tabpfn-mix-1.0-regressor model through api, including Node.js, Python, http.

tabpfn-mix-1.0-regressor huggingface.co Url

https://huggingface.co/autogluon/tabpfn-mix-1.0-regressor

autogluon tabpfn-mix-1.0-regressor online free

tabpfn-mix-1.0-regressor huggingface.co is an online trial and call api platform, which integrates tabpfn-mix-1.0-regressor's modeling effects, including api services, and provides a free online trial of tabpfn-mix-1.0-regressor, you can try tabpfn-mix-1.0-regressor online for free by clicking the link below.

autogluon tabpfn-mix-1.0-regressor online free url in huggingface.co:

https://huggingface.co/autogluon/tabpfn-mix-1.0-regressor

tabpfn-mix-1.0-regressor install

tabpfn-mix-1.0-regressor is an open source model from GitHub that offers a free installation service, and any user can find tabpfn-mix-1.0-regressor on GitHub to install. At the same time, huggingface.co provides the effect of tabpfn-mix-1.0-regressor install, users can directly use tabpfn-mix-1.0-regressor installed effect in huggingface.co for debugging and trial. It also supports api for free installation.

tabpfn-mix-1.0-regressor install url in huggingface.co:

https://huggingface.co/autogluon/tabpfn-mix-1.0-regressor

Url of tabpfn-mix-1.0-regressor

tabpfn-mix-1.0-regressor huggingface.co Url

Provider of tabpfn-mix-1.0-regressor huggingface.co

autogluon
ORGANIZATIONS

Other API from autogluon