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.018178 | 0.635922 | 0.018178 | 0.635922 | 1 | True | 5 |
1 | WeightedEnsemble_L2 | 0.83 | 0.018721 | 0.896239 | 0.000542 | 0.260318 | 2 | True | 13 |
2 | HierarchicalShrinkageTree_2 | 0.82 | 0.017990 | 1.044418 | 0.017990 | 1.044418 | 1 | True | 11 |
3 | HierarchicalShrinkageTree | 0.82 | 0.018742 | 1.035892 | 0.018742 | 1.035892 | 1 | True | 10 |
4 | HierarchicalShrinkageTree_3 | 0.82 | 0.021276 | 1.044300 | 0.021276 | 1.044300 | 1 | True | 12 |
5 | GreedyTree | 0.81 | 0.018385 | 0.719430 | 0.018385 | 0.719430 | 1 | True | 4 |
6 | BoostedRules_2 | 0.81 | 0.018939 | 0.643160 | 0.018939 | 0.643160 | 1 | True | 6 |
7 | RuleFit_3 | 0.77 | 0.079182 | 2.140148 | 0.079182 | 2.140148 | 1 | True | 3 |
8 | RuleFit | 0.73 | 0.044107 | 2.389099 | 0.044107 | 2.389099 | 1 | True | 1 |
9 | RuleFit_2 | 0.73 | 0.070148 | 2.072358 | 0.070148 | 2.072358 | 1 | True | 2 |
10 | Figs | 0.27 | 0.017628 | 0.648369 | 0.017628 | 0.648369 | 1 | True | 7 |
11 | Figs_2 | 0.27 | 0.017874 | 0.671452 | 0.017874 | 0.671452 | 1 | True | 8 |
12 | Figs_3 | 0.27 | 0.018246 | 0.692464 | 0.018246 | 0.692464 | 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-20230222_232840/models/Boos... | 0.635922 | 0.018178 | 0 | 2 | 2 | {'random_state': 0, 'n_estimators': 5} |
HierarchicalShrinkageTree | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230222_232840/models/Hier... | 1.035892 | 0.018742 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 6} |
HierarchicalShrinkageTree_2 | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230222_232840/models/Hier... | 1.044418 | 0.017990 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 12} |
HierarchicalShrinkageTree_3 | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230222_232840/models/Hier... | 1.044300 | 0.021276 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 18} |
BoostedRules_2 | BoostedRulesModel | 0.81 | 10.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230222_232840/models/Boos... | 0.643160 | 0.018939 | 0 | 2 | 2 | {'random_state': 0, 'n_estimators': 10} |
GreedyTree | GreedyTreeModel | 0.81 | 17.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230222_232840/models/Gree... | 0.719430 | 0.018385 | 0 | 2 | 2 | {'random_state': 0, 'max_leaf_nodes': 18} |
RuleFit_3 | RuleFitModel | 0.77 | 35.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230222_232840/models/Rule... | 2.140148 | 0.079182 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 18} |
RuleFit | RuleFitModel | 0.73 | 8.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230222_232840/models/Rule... | 2.389099 | 0.044107 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 7} |
RuleFit_2 | RuleFitModel | 0.73 | 29.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230222_232840/models/Rule... | 2.072358 | 0.070148 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 12} |
Figs | FigsModel | 0.27 | 6.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230222_232840/models/Figs/ | 0.648369 | 0.017628 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 6} |
Figs_2 | FigsModel | 0.27 | 10.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230222_232840/models/Figs_2/ | 0.671452 | 0.017874 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 10} |
Figs_3 | FigsModel | 0.27 | 15.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20230222_232840/models/Figs_3/ | 0.692464 | 0.018246 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 15} |