@@ -80,6 +80,12 @@ pca_order <- function(x, y) {
80
80
preprocess_predictors_fit <- function (processed ) {
81
81
predictors <- as.data.table(processed $ predictors )
82
82
83
+ # Convert logicals to binary integers
84
+ logical_cols <- names(which(sapply(predictors , is.logical )))
85
+ if (length(logical_cols ) > 0 ) {
86
+ predictors [, (logical_cols ) : = lapply(.SD , as.integer ), .SDcols = logical_cols ]
87
+ }
88
+
83
89
# Convert characters to factors
84
90
char_cols <- names(which(sapply(predictors , is.character )))
85
91
if (length(char_cols ) > 0 ) {
@@ -114,6 +120,12 @@ preprocess_predictors_fit <- function(processed) {
114
120
preprocess_predictors_predict <- function (object , predictors ) {
115
121
predictors <- as.data.table(predictors )
116
122
123
+ # Convert logicals to binary integers
124
+ logical_cols <- names(which(sapply(predictors , is.logical )))
125
+ if (length(logical_cols ) > 0 ) {
126
+ predictors [, (logical_cols ) : = lapply(.SD , as.integer ), .SDcols = logical_cols ]
127
+ }
128
+
117
129
# Convert characters to factors
118
130
char_cols <- names(which(sapply(predictors , is.character )))
119
131
if (length(char_cols ) > 0 ) {
@@ -152,7 +164,7 @@ preprocess_outcome <- function(processed, loss) {
152
164
153
165
if (is_binary & is_integerish ) {
154
166
warning(paste(
155
- " y is binary integer, assuming regression task." ,
167
+ " y is a binary integer, assuming regression task." ,
156
168
" Recode y to a factor for classification."
157
169
))
158
170
}
0 commit comments