Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement StableForestRegressor #17

Merged
merged 3 commits into from
Jun 15, 2023
Merged

Implement StableForestRegressor #17

merged 3 commits into from
Jun 15, 2023

Conversation

rikhuijzer
Copy link
Owner

@rikhuijzer rikhuijzer commented Jun 15, 2023

Follow-up on #16.

Works towards #13.

Results

13×7 DataFrame
 Row │ Dataset   Model                   Hyperparameters    `nfolds`  AUC     RMS     1.96*SE
     │ String    String                  String             Int64     String  String  String
─────┼────────────────────────────────────────────────────────────────────────────────────────
   1 │ blobs     LGBMClassifier          (;)                      10  0.99            0.01
   2 │ blobs     LGBMClassifier          (max_depth = 2,)         10  0.99            0.01
   3 │ blobs     StableRulesClassifier   (n_trees = 50,)          10  1.00            0.00
   4 │ titanic   LGBMClassifier          (;)                      10  0.87            0.03
   5 │ titanic   LGBMClassifier          (max_depth = 2,)         10  0.85            0.02
   6 │ titanic   StableForestClassifier  (n_trees = 1500,)        10  0.85            0.02
   7 │ titanic   StableRulesClassifier   (n_trees = 1500,)        10  0.83            0.02
   8 │ haberman  LGBMClassifier          (;)                      10  0.71            0.06
   9 │ haberman  LGBMClassifier          (max_depth = 2,)         10  0.67            0.05
  10 │ haberman  StableForestClassifier  (n_trees = 1500,)        10  0.70            0.05
  11 │ haberman  StableRulesClassifier   (n_trees = 1500,)        10  0.67            0.04
  12 │ boston    LinearRegressor         (;)                      10          0.70    0.05
  13 │ boston    StableForestRegressor   (;)                      10          0.66    0.07

@rikhuijzer rikhuijzer changed the title Implement regression (part 2) Implement StableForestRegressor Jun 15, 2023
@rikhuijzer rikhuijzer marked this pull request as ready for review June 15, 2023 10:26
@rikhuijzer rikhuijzer enabled auto-merge (squash) June 15, 2023 10:31
@rikhuijzer rikhuijzer merged commit 8c01f5e into main Jun 15, 2023
@rikhuijzer rikhuijzer deleted the rh/regression-part-2 branch June 15, 2023 10:50
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant