Is pretrained model available?

[hqng]

I had read the paper. It is very interesting idea and a step forward from TabPFN.
Now I am looking at the MotherNet API and would like to try it on my internal data. However, it is quite frustrated since I cannot found any pretrained model. Am I missing something? or can you open-source a pretrained model? Thank you so much.

[amueller]

Hey! sorry for the slow response. There's an example here:

mothernet/mothernet/tests/test_follow_sklearn_interface.py

Line 47 in 762f63b

def test_mothernet_paper():

Using

from mothernet.prediction import MotherNetClassifier
from mothernet.utils import get_mn_model
model_string = "mn_d2048_H4096_L2_W32_P512_1_gpu_warm_08_25_2023_21_46_25_epoch_3940_no_optimizer.pickle"
    model_path = get_mn_model(model_string)
classifier = MotherNetClassifier(path=model_path)

yields the model from the arXiv paper.

Note that this is still a WIP and see the paper for some of the failure cases. In particular results with categorical variables depend a lot on whether OneHotEncoding is used and useful for a particular dataset.
Note that MotherNetClassifier doesn't do any ensembling and you'd need to use EnsembleMeta to get the ensemble as in the paper.

[hqng]

Thank you for your information. I'll try it later.