# Load the iris dataset
data(iris)
# Create an imbalanced dataset by taking a subset
imbal_iris <- iris[c(1:40, 51:100, 101:110), ]
table(imbal_iris$Species) # Show class distribution
# Balance classes using the default "balance" strategy
balanced_iris <- smote_classif(Species ~ ., imbal_iris)
table(balanced_iris$Species) # Show balanced distribution
# Custom over/under-sampling
custom_iris <- smote_classif(Species ~ ., imbal_iris,
C.perc = list(setosa = 0.8,
versicolor = 1,
virginica = 3))
table(custom_iris$Species)
Run the code above in your browser using DataLab