search

pbiecek / ceterisParibus

Ceteris Paribus Plots (What-If plots) for explanations of a single observation
https://pbiecek.github.io/ceterisParibus/
41 stars 13 forks source link

Plot subset of variables #3

Closed agosiewska closed 6 years ago

agosiewska commented 6 years ago

It would be great if there was a possibility for generating plots for the subset of variables. I was using cp plots for the model with 35 variables and it was difficult to read anything.

I'm not sure if it's better to add such parameter to ceteris_paribus() or plot().

pbiecek commented 6 years ago

It is added to ceteris_paribus() as the selected_variables argument

wi_rf <- ceteris_paribus(explainer_rf, observation = new_apartment,
         selected_variables = c("surface", "floor", "no.rooms"))
agosiewska commented 6 years ago

Thank you