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 | WeightedEnsemble_L2 | 0.83 | 0.073246 | 2.104772 | 0.000538 | 0.267963 | 2 | True | 13 |
1 | HierarchicalShrinkageTree_2 | 0.82 | 0.009560 | 0.416293 | 0.009560 | 0.416293 | 1 | True | 11 |
2 | HierarchicalShrinkageTree_3 | 0.82 | 0.009768 | 0.415811 | 0.009768 | 0.415811 | 1 | True | 12 |
3 | HierarchicalShrinkageTree | 0.82 | 0.009960 | 0.415582 | 0.009960 | 0.415582 | 1 | True | 10 |
4 | Figs | 0.82 | 0.011085 | 0.044175 | 0.011085 | 0.044175 | 1 | True | 7 |
5 | BoostedRules_2 | 0.82 | 0.011199 | 0.029617 | 0.011199 | 0.029617 | 1 | True | 6 |
6 | GreedyTree | 0.81 | 0.009811 | 0.028647 | 0.009811 | 0.028647 | 1 | True | 4 |
7 | Figs_2 | 0.81 | 0.011391 | 0.069298 | 0.011391 | 0.069298 | 1 | True | 8 |
8 | BoostedRules | 0.78 | 0.010716 | 0.025449 | 0.010716 | 0.025449 | 1 | True | 5 |
9 | Figs_3 | 0.78 | 0.011952 | 0.105032 | 0.011952 | 0.105032 | 1 | True | 9 |
10 | RuleFit_3 | 0.77 | 0.062940 | 1.420998 | 0.062940 | 1.420998 | 1 | True | 3 |
11 | RuleFit | 0.73 | 0.029015 | 1.834227 | 0.029015 | 1.834227 | 1 | True | 1 |
12 | RuleFit_2 | 0.73 | 0.054874 | 1.401173 | 0.054874 | 1.401173 | 1 | True | 2 |
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 | |
---|---|---|---|---|---|---|---|---|---|---|---|
Figs | FigsModel | 0.82 | 6.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20220728_211141/models/Figs/ | 0.044175 | 0.011085 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 6} |
HierarchicalShrinkageTree | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20220728_211141/models/Hier... | 0.415582 | 0.009960 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 6} |
HierarchicalShrinkageTree_2 | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20220728_211141/models/Hier... | 0.416293 | 0.009560 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 12} |
HierarchicalShrinkageTree_3 | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20220728_211141/models/Hier... | 0.415811 | 0.009768 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 18} |
BoostedRules_2 | BoostedRulesModel | 0.82 | 20.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20220728_211141/models/Boos... | 0.029617 | 0.011199 | 0 | 2 | 2 | {'random_state': 0, 'n_estimators': 10} |
Figs_2 | FigsModel | 0.81 | 10.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20220728_211141/models/Figs_2/ | 0.069298 | 0.011391 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 10} |
GreedyTree | GreedyTreeModel | 0.81 | 17.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20220728_211141/models/Gree... | 0.028647 | 0.009811 | 0 | 2 | 2 | {'random_state': 0, 'max_leaf_nodes': 18} |
BoostedRules | BoostedRulesModel | 0.78 | 10.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20220728_211141/models/Boos... | 0.025449 | 0.010716 | 0 | 2 | 2 | {'random_state': 0, 'n_estimators': 5} |
Figs_3 | FigsModel | 0.78 | 15.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20220728_211141/models/Figs_3/ | 0.105032 | 0.011952 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 15} |
RuleFit_3 | RuleFitModel | 0.77 | 35.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20220728_211141/models/Rule... | 1.420998 | 0.062940 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 18} |
RuleFit | RuleFitModel | 0.73 | 8.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20220728_211141/models/Rule... | 1.834227 | 0.029015 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 7} |
RuleFit_2 | RuleFitModel | 0.73 | 29.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20220728_211141/models/Rule... | 1.401173 | 0.054874 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 12} |