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.019706 | 0.635678 | 0.019706 | 0.635678 | 1 | True | 5 |
1 | WeightedEnsemble_L2 | 0.83 | 0.020289 | 0.906906 | 0.000583 | 0.271228 | 2 | True | 13 |
2 | HierarchicalShrinkageTree_3 | 0.82 | 0.018290 | 1.136698 | 0.018290 | 1.136698 | 1 | True | 12 |
3 | HierarchicalShrinkageTree | 0.82 | 0.019066 | 1.048132 | 0.019066 | 1.048132 | 1 | True | 10 |
4 | HierarchicalShrinkageTree_2 | 0.82 | 0.019331 | 1.052087 | 0.019331 | 1.052087 | 1 | True | 11 |
5 | GreedyTree | 0.81 | 0.018659 | 0.634258 | 0.018659 | 0.634258 | 1 | True | 4 |
6 | BoostedRules_2 | 0.81 | 0.020419 | 0.645071 | 0.020419 | 0.645071 | 1 | True | 6 |
7 | RuleFit_3 | 0.77 | 0.080528 | 2.092422 | 0.080528 | 2.092422 | 1 | True | 3 |
8 | RuleFit | 0.73 | 0.044827 | 2.265908 | 0.044827 | 2.265908 | 1 | True | 1 |
9 | RuleFit_2 | 0.73 | 0.072741 | 2.089416 | 0.072741 | 2.089416 | 1 | True | 2 |
10 | Figs | 0.27 | 0.018914 | 0.662505 | 0.018914 | 0.662505 | 1 | True | 7 |
11 | Figs_3 | 0.27 | 0.019456 | 0.732935 | 0.019456 | 0.732935 | 1 | True | 9 |
12 | Figs_2 | 0.27 | 0.019548 | 0.688853 | 0.019548 | 0.688853 | 1 | True | 8 |
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-20230111_022240/models/Boos... | 0.635678 | 0.019706 | 0 | 2 | 2 | {'random_state': 0, 'n_estimators': 5} |
HierarchicalShrinkageTree | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230111_022240/models/Hier... | 1.048132 | 0.019066 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 6} |
HierarchicalShrinkageTree_2 | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230111_022240/models/Hier... | 1.052087 | 0.019331 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 12} |
HierarchicalShrinkageTree_3 | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230111_022240/models/Hier... | 1.136698 | 0.018290 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 18} |
BoostedRules_2 | BoostedRulesModel | 0.81 | 10.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230111_022240/models/Boos... | 0.645071 | 0.020419 | 0 | 2 | 2 | {'random_state': 0, 'n_estimators': 10} |
GreedyTree | GreedyTreeModel | 0.81 | 17.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230111_022240/models/Gree... | 0.634258 | 0.018659 | 0 | 2 | 2 | {'random_state': 0, 'max_leaf_nodes': 18} |
RuleFit_3 | RuleFitModel | 0.77 | 35.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230111_022240/models/Rule... | 2.092422 | 0.080528 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 18} |
RuleFit | RuleFitModel | 0.73 | 8.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230111_022240/models/Rule... | 2.265908 | 0.044827 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 7} |
RuleFit_2 | RuleFitModel | 0.73 | 29.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230111_022240/models/Rule... | 2.089416 | 0.072741 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 12} |
Figs | FigsModel | 0.27 | 6.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230111_022240/models/Figs/ | 0.662505 | 0.018914 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 6} |
Figs_2 | FigsModel | 0.27 | 10.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230111_022240/models/Figs_2/ | 0.688853 | 0.019548 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 10} |
Figs_3 | FigsModel | 0.27 | 15.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230111_022240/models/Figs_3/ | 0.732935 | 0.019456 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 15} |