DistrictDataLabs / yellowbrick

Visual analysis and diagnostic tools to facilitate machine learning model selection.
http://www.scikit-yb.org/
Apache License 2.0
4.24k stars 555 forks source link

Tree-Depth Plots #305

Open bbengfort opened 6 years ago

bbengfort commented 6 years ago

For random forest hyperparameter tuning.

See Finding Similar Days for Air Traffic Management for examples.

Create a parameter to set the y-axis to either concordance index or brier score:

http://scikit-learn.org/stable/modules/generated/sklearn.metrics.brier_score_loss.html https://github.com/sebp/scikit-survival/blob/master/sksurv/metrics.py

bbengfort commented 6 years ago

Note: create a yellowbrick.tree package specifically for tree visualizers. See also #194

bbengfort commented 6 years ago
screenshot 2018-03-21 23 26 03 screenshot 2018-03-21 23 26 15 screenshot 2018-03-21 23 26 30