EconML icon indicating copy to clipboard operation
EconML copied to clipboard

Support get the data for plotting.

Open MimeZoe0628 opened this issue 2 years ago • 1 comments

I use SingleTreePolicyInterpreter to interpret the estimator and save as a PNG file. The image like this: policy

I want to display it on html by using the third-party library (e.g D3.js, Echart) in an interactive way. But I can not get the data to plot without modifing EconML source code. There is a need to get the relevant attribute to use for plotting.

MimeZoe0628 avatar Feb 24 '23 14:02 MimeZoe0628

Could you expand on exactly what data you'd like us to output? In particular, how does what you want differ from what the treat method on the class provides? That method takes a set of features and returns the treatment suggested by the tree for each row.

kbattocchi avatar Feb 24 '23 14:02 kbattocchi