This function uses Tomek links to perform under-sampling for handling imbalanced multiclass problems. Tomek links are broken by removing one or both examples forming the link.
A formula describing the prediction problem.
A data frame containing the original imbalanced data set.
A character string indicating which distance metric to use when determining the k nearest neighbors. See the details. Defaults to "Euclidean".
A number indicating the value of p if the "p-norm" distance is chosen. Only necessary to define if a "p-norm" is chosen in the
A character vector indicating which classes should be under-sampled. Defaults to "all" meaning that examples from all existing classes can be removed. The user may also specify a subset of classes for which tomek links should be removed.
A character string indicating if both examples forming the Tomek link are to be removed, or if only the example from the larger class should be discarded. In the first case this parameter should be set to "both" and in the second case should be set to "maj".
dist allows the user to define the distance metric to be used in the neighbors computation. Although the default is the Euclidean distance, other metrics are available. This allows the computation of distances in data sets with, for instance, both nominal and numeric features. The options available for the distance functions are as follows:
- for data with only numeric features: "Manhattan", "Euclidean", "Canberra", "Chebyshev", "p-norm";
- for data with only nominal features: "Overlap";
- for dealing with both nominal and numeric features: "HEOM", "HVDM".
When the "p-norm" is selected for the
dist parameter, it is also necessary to define the value of parameter
p. The value of parameter
p sets which "p-norm" will be used. For instance, if
p is set to 1, the "1-norm" (or Manhattan distance) is used, and if
p is set to 2, the "2-norm" (or Euclidean distance) is applied.
For more details regarding the distance functions implemented in UBL package please see the package vignettes.
This function performs an under-sampling strategy based on the notion of Tomek links for imbalanced multiclass problems. Two examples form a Tomek link if they are each other closest neighbors and they have different class labels.
The under-sampling procedure can be performed in two different ways. When detected the Tomek links, the examples of both classes can be removed, or the Tomek link can be broken by removing only one of the examples (traditionally the one belonging to the majority class). This function also includes these two procedures. Moreover, it allows for the user to identify in which classes under-sampling should be applied. These two aspects are controlled by the
rem parameters. The
Cl parameter is used to express the classes that can be under-sampled and its default is "all" (all existing classes are candidates for having examples removed). The parameter
rem indicates if the Tomek link is broken by removing both examples ("both") or by removing only the example belonging to the more populated class between the two existing in the Tomek link.
Note that the options for
rem may "disagree". In those cases, the preference is given to the
Cl options once the user choose that specific set of classes to under-sample and not the other ones (even if the defined classes are not the larger ones). This means that, when making a decision on how many and which examples will be removed the first criteria used will be the
Cl definition .
For a better clarification of the impact of the options selected for Cl and rem parameters we now provide some possible scenarios and the expected behavior:
Cl is set to one class which is neither the more nor the less frequent, and
rem is set to "maj". The expected behavior is the following:
- if a Tomek link exists connecting the largest class and another class(not included in
Cl): no example is removed;
- if a Tomek link exists connecting the larger class and the class defined in
Cl: the example from the
Cl class is removed (because the user expressly indicates that only examples from class
Cl should be removed);
Cl includes two classes and
rem is set to "both". This function will do the following:
- if a Tomek link exists between an example with class in
Cl and another example with class not in
Cl, then, only the example with class in
Cl is removed;
- if the Tomek link exists between two examples with classes in
Cl, then, both are removed.
Cl includes two classes and
rem is set to "maj". The behavior of this function is the following:
-if a Tomek link exists connecting two classes included in
Cl, then only the example belonging to the more populated class is removed;
-if a Tomek link exists connecting an example from a class included in
Cl and another example whose class is not in
Cl and is the largest class, then, no example is removed.
The function returns a list containing a data frame with the new data set resulting from the application of the Tomek link strategy defined, and the indexes of the examples removed.
Tomek, I. (1976). Two modifications of CNN IEEE Trans. Syst. Man Cybern., 769-772
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
library(DMwR) data(algae) clean.algae <- algae[complete.cases(algae), ] alg.HVDM1 <- TomekClassif(season~., clean.algae, dist = "HVDM", Cl = c("winter", "spring"), rem = "both") alg.HVDM2 <- TomekClassif(season~., clean.algae, dist = "HVDM", rem = "maj") # removes only examples from class summer which are the # majority class in the link alg.EuM <- TomekClassif(season~., clean.algae, dist = "HEOM", Cl = "summer", rem = "maj") # removes only examples from class summer in every link they appear alg.EuB <- TomekClassif(season~., clean.algae, dist = "HEOM", Cl = "summer", rem = "both") summary(clean.algae$season) summary(alg.HVDM1[]$season) summary(alg.HVDM2[]$season) summary(alg.EuM[]$season) summary(alg.EuB[]$season) # check which were the indexes of the examples removed in alg.EuM alg.EuM[]
Loading required package: MBA Loading required package: gstat Loading required package: automap Loading required package: sp Loading required package: randomForest randomForest 4.6-12 Type rfNews() to see new features/changes/bug fixes. Loading required package: lattice Loading required package: grid autumn spring summer winter 36 48 43 57 autumn spring summer winter 36 31 43 36 autumn spring summer winter 36 43 37 36 autumn spring summer winter 36 48 37 57 autumn spring summer winter 36 48 27 57  7 35 75 95 165 180
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.