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.099216 | 3.479751 | 0.000589 | 0.285408 | 2 | True | 13 |
1 | HierarchicalShrinkageTree_3 | 0.82 | 0.018755 | 1.133771 | 0.018755 | 1.133771 | 1 | True | 12 |
2 | Figs | 0.82 | 0.019264 | 0.659746 | 0.019264 | 0.659746 | 1 | True | 7 |
3 | BoostedRules_2 | 0.82 | 0.019817 | 0.639075 | 0.019817 | 0.639075 | 1 | True | 6 |
4 | HierarchicalShrinkageTree | 0.82 | 0.021195 | 1.048791 | 0.021195 | 1.048791 | 1 | True | 10 |
5 | HierarchicalShrinkageTree_2 | 0.82 | 0.021696 | 1.046431 | 0.021696 | 1.046431 | 1 | True | 11 |
6 | GreedyTree | 0.81 | 0.018515 | 0.634493 | 0.018515 | 0.634493 | 1 | True | 4 |
7 | Figs_2 | 0.81 | 0.018829 | 0.691391 | 0.018829 | 0.691391 | 1 | True | 8 |
8 | BoostedRules | 0.78 | 0.018710 | 0.635131 | 0.018710 | 0.635131 | 1 | True | 5 |
9 | Figs_3 | 0.78 | 0.019958 | 0.733970 | 0.019958 | 0.733970 | 1 | True | 9 |
10 | RuleFit_3 | 0.77 | 0.079872 | 2.060572 | 0.079872 | 2.060572 | 1 | True | 3 |
11 | RuleFit | 0.73 | 0.042912 | 2.236065 | 0.042912 | 2.236065 | 1 | True | 1 |
12 | RuleFit_2 | 0.73 | 0.080854 | 2.108514 | 0.080854 | 2.108514 | 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-20221213_014805/models/Figs/ | 0.659746 | 0.019264 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 6} |
HierarchicalShrinkageTree | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20221213_014805/models/Hier... | 1.048791 | 0.021195 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 6} |
HierarchicalShrinkageTree_2 | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20221213_014805/models/Hier... | 1.046431 | 0.021696 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 12} |
HierarchicalShrinkageTree_3 | HSTreeModel | 0.82 | 19.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20221213_014805/models/Hier... | 1.133771 | 0.018755 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 18} |
BoostedRules_2 | BoostedRulesModel | 0.82 | 20.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20221213_014805/models/Boos... | 0.639075 | 0.019817 | 0 | 2 | 2 | {'random_state': 0, 'n_estimators': 10} |
Figs_2 | FigsModel | 0.81 | 10.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20221213_014805/models/Figs_2/ | 0.691391 | 0.018829 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 10} |
GreedyTree | GreedyTreeModel | 0.81 | 17.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20221213_014805/models/Gree... | 0.634493 | 0.018515 | 0 | 2 | 2 | {'random_state': 0, 'max_leaf_nodes': 18} |
BoostedRules | BoostedRulesModel | 0.78 | 10.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20221213_014805/models/Boos... | 0.635131 | 0.018710 | 0 | 2 | 2 | {'random_state': 0, 'n_estimators': 5} |
Figs_3 | FigsModel | 0.78 | 15.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20221213_014805/models/Figs_3/ | 0.733970 | 0.019958 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 15} |
RuleFit_3 | RuleFitModel | 0.77 | 35.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20221213_014805/models/Rule... | 2.060572 | 0.079872 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 18} |
RuleFit | RuleFitModel | 0.73 | 8.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20221213_014805/models/Rule... | 2.236065 | 0.042912 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 7} |
RuleFit_2 | RuleFitModel | 0.73 | 29.0 | WeightedEnsemble_L2 | AutogluonModels/ag-20221213_014805/models/Rule... | 2.108514 | 0.080854 | 0 | 2 | 2 | {'random_state': 0, 'max_rules': 12} |