Tfdf: R2 value, and SHAP value plot?

TensorFlow General Discussion
,
1

For the tfdf RandomForestModel regression model, is there any way to generate R2 value, and SHAP value plot?

3

Hi Frank,

Keras does not have a native R2 score implementation (see the list of Keras’ regression metrics). However, the R2 score can be computed using a custom metric. See an example.

TF-DF does not have a native TreeSHAP implementation (yet? :slight_smile:). If this is of interest, please create a feature request here so we can prioritize new features. In the mean time, I would use a model agnostic implementation of SHAP.

Cheers,
M.

2 Likes
5

Thank you, Mathieu! I will create a feature request for SHAP plot.

1 Like