search

automl / TabPFN

Official implementation of the TabPFN paper (https://arxiv.org/abs/2207.01848) and the tabpfn package.
http://priorlabs.ai
Apache License 2.0
1.22k stars 109 forks source link

How to disable overwrite_warning=True without putting it in fit #52

Closed apavlo89 closed 1 year ago

apavlo89 commented 1 year ago

Is there a way to disable this warning as I want to use TabPFN as a stacking estimator (using TPOT's easy stacking estimator functionality). I've tried a bunch of ways but I think the only is to disable this bs from TabPFN script.. Any advice on how to do that?

noahho commented 1 year ago 
from functools import partial
classifier = TabPFNClassifier(device=device, N_ensemble_configurations=16)
classifier.fit = partial(classifier.fit, overwrite_warning=True)