filtering example

ModelOriented · Dec 18, 2018 · 098c450 · 098c450
1 parent 5da4273
commit 098c450
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -88,8 +88,7 @@ plot(ive_rf)
 ```
 # filtered
 ive_rf_filtered <- dplyr::filter(ive_rf, `_ylevel_` =="fired")
-class(ive_rf_filtered) <- c("individual_variable_effect", "data.frame")
-plot(ive_rf_filtered)
+shapper:::plot.individual_variable_effect(ive_rf_filtered)
 ```
 
 <img src="materials/classification_plot_filtered.png" width="100%" /> 

diff --git a/vignettes/shapper_classification.Rmd b/vignettes/shapper_classification.Rmd
@@ -118,6 +118,5 @@ plot(ive_rf, ive_tree, id = c(1,2), show_predcited = FALSE,
 
 ```{r}
 ive_rf_filtered <- dplyr::filter(ive_rf, `_ylevel_` =="fired")
-class(ive_rf_filtered) <- c("individual_variable_effect", "data.frame")
-plot(ive_rf_filtered)
+shapper:::plot.individual_variable_effect(ive_rf_filtered)
 ```