search

autogluon / tabrepo

Apache License 2.0
36 stars 9 forks source link

Sparse #54

Closed geoalgo closed 7 months ago

geoalgo commented 10 months ago

This adds logic to fallback to extra-trees in case of missing predictions. I have added test that make sure that the expected behavior happens for tabular predictions and the repository. For runtimes, the current logic will impute the mean of model runtime if it is not found.

The logic _keep_only_models_in_both_validation_and_test only keeps model which are present in both val and test split (this was just assumed and asserted in previous version) now the filter happens automatically.

geoalgo commented 10 months ago

Note: the tests are failing but I will only be able to update the change by tomorrow. However, you may still want to take a look as most of the logic is there.