CCS-Lab / easyml

A toolkit for easily building and evaluating machine learning models.
https://ccs-lab.github.io/easyml
Other
40 stars 16 forks source link

Random forest on a continuous dependent measure --> replace RMSE with correlation coefficients #75

Closed youngahn closed 7 years ago

youngahn commented 7 years ago

With easy_random_forest, currently I see histograms of RMSE across X number of training/test divisions. Just like easy_glmnet, it should plot histograms of correlation coefficients (perhaps along with RMSE?).

Nathaniel-Haines commented 7 years ago

The easy_random_forest() function will output histograms of R^2 metrics if the measure argument is set to measure_r2_score (i.e. measure = measure_r2_score). If this is made the default, I think it would solve the issue.

paulhendricks commented 7 years ago

See e94013dc8bee9a8f8189b14669ba8713555c4646.