Skip to content

Commit

Permalink
chore: styling
Browse files Browse the repository at this point in the history
  • Loading branch information
egillax committed Dec 23, 2023
1 parent 1ea7d5f commit d2469f6
Show file tree
Hide file tree
Showing 7 changed files with 301 additions and 197 deletions.
9 changes: 6 additions & 3 deletions R/Dataset.R
Original file line number Diff line number Diff line change
Expand Up @@ -23,13 +23,16 @@ createDataset <- function(data, labels, plpModel = NULL) {
attributes(data)$path <- attributes(data)$dbname
}
if (is.null(plpModel)) {
data <- dataset(r_to_py(normalizePath(attributes(data)$path)),
r_to_py(labels$outcomeCount))
data <- dataset(
r_to_py(normalizePath(attributes(data)$path)),
r_to_py(labels$outcomeCount)
)
} else {
numericalFeatures <-
r_to_py(as.array(which(plpModel$covariateImportance$isNumeric)))
data <- dataset(r_to_py(normalizePath(attributes(data)$path)),
numerical_features = numericalFeatures)
numerical_features = numericalFeatures
)
}

return(data)
Expand Down
Loading

0 comments on commit d2469f6

Please sign in to comment.