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

Run experiments tuning random forest params #76

Closed
wants to merge 1 commit into from

Conversation

dberenbaum
Copy link
Collaborator

Better RF split and number of estimators based on small grid search.

@github-actions
Copy link

CML Report

Plots

ROC
Precision-Recall
Confusion Matrix

Feature Importance: main

Feature Importance: main

Feature Importance: workspace

Feature Importance: workspace

Metrics and Params

main → workspace

Path Metric main workspace Change
eval/metrics.json avg_prec.test 0.925 0.92688 0.0018818
eval/metrics.json avg_prec.train 0.97437 0.99932 0.024952
eval/metrics.json roc_auc.test 0.94602 0.94509 -0.00092798
eval/metrics.json roc_auc.train 0.98667 0.99971 0.013041
Path Param main workspace Change
params.yaml train.min_split 0.01 8 7.99
params.yaml train.n_est 50 100 50

@dberenbaum dberenbaum closed this Apr 19, 2024
@dberenbaum dberenbaum deleted the tune-hyperparams branch April 19, 2024 17: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
1 participant