age | workclass | fnlwgt | education | education-num | marital-status | occupation | relationship | race | sex | capital-gain | capital-loss | hours-per-week | native-country | class | |
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
6118 | 51 | Private | 39264 | Some-college | 10 | Married-civ-spouse | Exec-managerial | Wife | White | Female | 0 | 0 | 40 | United-States | >50K |
23204 | 58 | Private | 51662 | 10th | 6 | Married-civ-spouse | Other-service | Wife | White | Female | 0 | 0 | 8 | United-States | <=50K |
29590 | 40 | Private | 326310 | Some-college | 10 | Married-civ-spouse | Craft-repair | Husband | White | Male | 0 | 0 | 44 | United-States | <=50K |
18116 | 37 | Private | 222450 | HS-grad | 9 | Never-married | Sales | Not-in-family | White | Male | 0 | 2339 | 40 | El-Salvador | <=50K |
33964 | 62 | Private | 109190 | Bachelors | 13 | Married-civ-spouse | Exec-managerial | Husband | White | Male | 15024 | 0 | 40 | United-States | >50K |
model | score_val | pred_time_val | fit_time | pred_time_val_marginal | fit_time_marginal | stack_level | can_infer | fit_order | |
---|---|---|---|---|---|---|---|---|---|
0 | BoostedRules | 0.83 | 0.019180 | 0.632811 | 0.019180 | 0.632811 | 1 | True | 5 |
1 | WeightedEnsemble_L2 | 0.83 | 0.019806 | 0.908838 | 0.000626 | 0.276026 | 2 | True | 13 |
2 | HierarchicalShrinkageTree_2 | 0.82 | 0.020184 | 1.042527 | 0.020184 | 1.042527 | 1 | True | 11 |
3 | HierarchicalShrinkageTree_3 | 0.82 | 0.020472 | 1.038884 | 0.020472 | 1.038884 | 1 | True | 12 |
4 | HierarchicalShrinkageTree | 0.82 | 0.021567 | 1.042077 | 0.021567 | 1.042077 | 1 | True | 10 |
5 | GreedyTree | 0.81 | 0.017851 | 0.633729 | 0.017851 | 0.633729 | 1 | True | 4 |
6 | BoostedRules_2 | 0.81 | 0.020172 | 0.640239 | 0.020172 | 0.640239 | 1 | True | 6 |
7 | RuleFit_3 | 0.77 | 0.076504 | 2.027355 | 0.076504 | 2.027355 | 1 | True | 3 |
8 | RuleFit | 0.73 | 0.043106 | 2.247696 | 0.043106 | 2.247696 | 1 | True | 1 |
9 | RuleFit_2 | 0.73 | 0.068908 | 2.007573 | 0.068908 | 2.007573 | 1 | True | 2 |
10 | Figs_2 | 0.27 | 0.018962 | 0.689084 | 0.018962 | 0.689084 | 1 | True | 8 |
11 | Figs | 0.27 | 0.019176 | 0.661822 | 0.019176 | 0.661822 | 1 | True | 7 |
12 | Figs_3 | 0.27 | 0.019487 | 0.736320 | 0.019487 | 0.736320 | 1 | True | 9 |
model_types | model_performance | complexity | model_best | model_paths | model_fit_times | model_pred_times | num_bag_folds | max_stack_level | num_classes | model_hyperparams | |
---|---|---|---|---|---|---|---|---|---|---|---|
BoostedRules | BoostedRulesModel | 0.83 | 5.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230204_010744/models/Boos... | 0.632811 | 0.019180 | 0 | 2 | 2 | {'random_state': 0, 'n_estimators': 5} |
HierarchicalShrinkageTree | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230204_010744/models/Hier... | 1.042077 | 0.021567 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 6} |
HierarchicalShrinkageTree_2 | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230204_010744/models/Hier... | 1.042527 | 0.020184 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 12} |
HierarchicalShrinkageTree_3 | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230204_010744/models/Hier... | 1.038884 | 0.020472 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 18} |
BoostedRules_2 | BoostedRulesModel | 0.81 | 10.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230204_010744/models/Boos... | 0.640239 | 0.020172 | 0 | 2 | 2 | {'random_state': 0, 'n_estimators': 10} |
GreedyTree | GreedyTreeModel | 0.81 | 17.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230204_010744/models/Gree... | 0.633729 | 0.017851 | 0 | 2 | 2 | {'random_state': 0, 'max_leaf_nodes': 18} |
RuleFit_3 | RuleFitModel | 0.77 | 35.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230204_010744/models/Rule... | 2.027355 | 0.076504 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 18} |
RuleFit | RuleFitModel | 0.73 | 8.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230204_010744/models/Rule... | 2.247696 | 0.043106 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 7} |
RuleFit_2 | RuleFitModel | 0.73 | 29.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230204_010744/models/Rule... | 2.007573 | 0.068908 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 12} |
Figs | FigsModel | 0.27 | 6.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230204_010744/models/Figs/ | 0.661822 | 0.019176 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 6} |
Figs_2 | FigsModel | 0.27 | 10.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230204_010744/models/Figs_2/ | 0.689084 | 0.018962 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 10} |
Figs_3 | FigsModel | 0.27 | 15.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230204_010744/models/Figs_3/ | 0.736320 | 0.019487 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 15} |