Added tags for models that only accept categorical predictors (kinda …

…related to issue #329)
topepo · Nov 30, 2015 · f3e3d70 · f3e3d70
1 parent b743a30
commit f3e3d70
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/models/files/awnb.R b/models/files/awnb.R
@@ -31,5 +31,5 @@ modelInfo <- list(label = "Naive Bayes Classifier with Attribute Weighting",
                   },
                   levels = function(x) x$obsLevels,
                   predictors = function(x, s = NULL, ...) x$xNames,
-                  tags = c("Bayesian Model"),
+                  tags = c("Bayesian Model", "Categorical Predictors Only"),
                   sort = function(x) x[order(x[,1]),])
diff --git a/models/files/awtan.R b/models/files/awtan.R
@@ -27,5 +27,5 @@ modelInfo <- list(label = "Tree Augmented Naive Bayes Classifier with Attribute
                   },
                   levels = function(x) x$obsLevels,
                   predictors = function(x, s = NULL, ...) x$xNames,
-                  tags = c("Bayesian Model"),
+                  tags = c("Bayesian Model", "Categorical Predictors Only"),
                   sort = function(x) x[order(x[,1]),])
diff --git a/models/files/nbDiscrete.R b/models/files/nbDiscrete.R
@@ -30,5 +30,5 @@ modelInfo <- list(label = "Naive Bayes Classifier",
                   },
                   levels = function(x) x$obsLevels,
                   predictors = function(x, s = NULL, ...) x$xNames,
-                  tags = c("Bayesian Model"),
+                  tags = c("Bayesian Model", "Categorical Predictors Only"),
                   sort = function(x) x[order(x[,1]),])
diff --git a/models/files/nbSearch.R b/models/files/nbSearch.R
@@ -48,5 +48,5 @@ modelInfo <- list(label = "Semi-Naive Structure Learner Wrapper",
                   },
                   levels = function(x) x$obsLevels,
                   predictors = function(x, s = NULL, ...) x$xNames,
-                  tags = c("Bayesian Model"),
+                  tags = c("Bayesian Model", "Categorical Predictors Only"),
                   sort = function(x) x[order(x[,1]),])
diff --git a/models/files/tan.R b/models/files/tan.R
@@ -34,5 +34,5 @@ modelInfo <- list(label = "Tree Augmented Naive Bayes Classifier",
                   },
                   levels = function(x) x$obsLevels,
                   predictors = function(x, s = NULL, ...) x$xNames,
-                  tags = c("Bayesian Model"),
+                  tags = c("Bayesian Model", "Categorical Predictors Only"),
                   sort = function(x) x[order(x[,1]),])
diff --git a/models/files/tanSearch.R b/models/files/tanSearch.R
@@ -49,5 +49,5 @@ modelInfo <- list(label = "Tree Augmented Naive Bayes Classifier Structure Learn
                   },
                   levels = function(x) x$obsLevels,
                   predictors = function(x, s = NULL, ...) x$xNames,
-                  tags = c("Bayesian Model"),
+                  tags = c("Bayesian Model", "Categorical Predictors Only"),
                   sort = function(x) x[order(x[,1]),])