[X] I provided code that demonstrates a minimal reproducible example.
[ ] I confirmed bug exists on the latest mainline of AutoGluon via source install.
[X] I confirmed bug exists on the latest stable version of AutoGluon.
Describe the bug
When using TabularPredictor with problem_type='multiclass' and fitting with the parameter presets='interpretable' it gives the following error: AssertionError: Unknown y_pred_proba format for problem_type="multiclass".
Expected behavior
The execution should finish correctly or an error message should be printed notifying the user that the preset interpretable is not available for problem_type multi class
To Reproduce
from autogluon.tabular import TabularPredictor, TabularDataset
import pandas as pd
N_SUBSAMPLE = 500 # subsample datasets for faster demo
N_TEST = 50
train_data = TabularDataset('https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv') # can be local CSV file as well, returns Pandas DataFrame
train_data = train_data.sample(N_SUBSAMPLE, random_state=0)
test_data = TabularDataset('https://autogluon.s3.amazonaws.com/datasets/Inc/test.csv') # another Pandas DataFrame
test_data = test_data.sample(N_TEST, random_state=0)
label = 'relationship'
y_train = train_data[label]
y_test = test_data[label]
X_train = pd.DataFrame(train_data.drop(columns=[label]))
X_test = pd.DataFrame(test_data.drop(columns=[label]))
predictor_multi = TabularPredictor(label=label, problem_type='multiclass').fit(train_data, presets='interpretable', time_limit=20)
Bug Report Checklist
Describe the bug
When using TabularPredictor with problem_type='multiclass' and fitting with the parameter presets='interpretable' it gives the following error: AssertionError: Unknown y_pred_proba format for
problem_type="multiclass"
.Expected behavior
The execution should finish correctly or an error message should be printed notifying the user that the preset interpretable is not available for problem_type multi class
To Reproduce
Screenshots / Logs
Installed Versions