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 #83

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

dberenbaum
Copy link

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

@shcheklein

This comment was marked as outdated.

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

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.

2 participants