gini_impurities {mltools} | R Documentation |
Identify group weighted gini impurities using pairs of columns within a dataset. Can be used to located hierarchical data, or 1-1 correspondences
gini_impurities(dt, wide = FALSE, verbose = FALSE)
dt |
A data.table with at least two columns |
wide |
Should the results be in wide format? |
verbose |
Should progress be printed to the screen? |
For pairs of columns (Var1, Var2) in a dataset, calculates the weighted gini impurity of Var2 relative to the groups determined by Var1
library(data.table)
gini_impurities(alien.train)
gini_impurities(alien.train, wide=TRUE)