Skip to content

Commit

Permalink
fix bug (#630)
Browse files Browse the repository at this point in the history
  • Loading branch information
szilard authored and guolinke committed Jun 17, 2017
1 parent 6fa5147 commit 03ba730
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 22 deletions.
22 changes: 11 additions & 11 deletions R-package/R/lgb.prepare_rules.R
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ lgb.prepare_rules <- function(data, rules = NULL) {

# Map characters/factors
is_fix <- which(list_classes %in% c("character", "factor"))
ruleset <- list()
rules <- list()

# Need to create rules?
if (length(is_fix) > 0) {
Expand All @@ -113,13 +113,13 @@ lgb.prepare_rules <- function(data, rules = NULL) {
mini_numeric <- as.numeric(mini_unique) # No respect of ordinality
}

# Create ruleset
# Create rules
indexed <- colnames(data)[i] # Index value
ruleset[[indexed]] <- mini_numeric # Numeric content
names(ruleset[[indexed]]) <- mini_unique # Character equivalent
rules[[indexed]] <- mini_numeric # Numeric content
names(rules[[indexed]]) <- mini_unique # Character equivalent

# Apply to real data column
set(data, j = i, value = unname(ruleset[[indexed]][mini_data]))
set(data, j = i, value = unname(rules[[indexed]][mini_data]))

}

Expand Down Expand Up @@ -150,7 +150,7 @@ lgb.prepare_rules <- function(data, rules = NULL) {

# Map characters/factors
is_fix <- which(list_classes %in% c("character", "factor"))
ruleset <- list()
rules <- list()

# Need to create rules?
if (length(is_fix) > 0) {
Expand All @@ -171,13 +171,13 @@ lgb.prepare_rules <- function(data, rules = NULL) {
mini_numeric <- as.numeric(mini_unique) # No respect of ordinality
}

# Create ruleset
# Create rules
indexed <- colnames(data)[i] # Index value
ruleset[[indexed]] <- mini_numeric # Numeric content
names(ruleset[[indexed]]) <- mini_unique # Character equivalent
rules[[indexed]] <- mini_numeric # Numeric content
names(rules[[indexed]]) <- mini_unique # Character equivalent

# Apply to real data column
data[[i]] <- unname(ruleset[[indexed]][mini_data])
data[[i]] <- unname(rules[[indexed]][mini_data])

}

Expand All @@ -194,6 +194,6 @@ lgb.prepare_rules <- function(data, rules = NULL) {

}

return(list(data = data, rules = ruleset))
return(list(data = data, rules = rules))

}
22 changes: 11 additions & 11 deletions R-package/R/lgb.prepare_rules2.R
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ lgb.prepare_rules2 <- function(data, rules = NULL) {

# Map characters/factors
is_fix <- which(list_classes %in% c("character", "factor"))
ruleset <- list()
rules <- list()

# Need to create rules?
if (length(is_fix) > 0) {
Expand All @@ -112,13 +112,13 @@ lgb.prepare_rules2 <- function(data, rules = NULL) {
mini_numeric <- as.integer(mini_unique) # No respect of ordinality
}

# Create ruleset
# Create rules
indexed <- colnames(data)[i] # Index value
ruleset[[indexed]] <- mini_numeric # Numeric content
names(ruleset[[indexed]]) <- mini_unique # Character equivalent
rules[[indexed]] <- mini_numeric # Numeric content
names(rules[[indexed]]) <- mini_unique # Character equivalent

# Apply to real data column
set(data, j = i, value = unname(ruleset[[indexed]][mini_data]))
set(data, j = i, value = unname(rules[[indexed]][mini_data]))

}

Expand Down Expand Up @@ -149,7 +149,7 @@ lgb.prepare_rules2 <- function(data, rules = NULL) {

# Map characters/factors
is_fix <- which(list_classes %in% c("character", "factor"))
ruleset <- list()
rules <- list()

# Need to create rules?
if (length(is_fix) > 0) {
Expand All @@ -169,13 +169,13 @@ lgb.prepare_rules2 <- function(data, rules = NULL) {
mini_numeric <- as.integer(mini_unique) # No respect of ordinality
}

# Create ruleset
# Create rules
indexed <- colnames(data)[i] # Index value
ruleset[[indexed]] <- mini_numeric # Numeric content
names(ruleset[[indexed]]) <- mini_unique # Character equivalent
rules[[indexed]] <- mini_numeric # Numeric content
names(rules[[indexed]]) <- mini_unique # Character equivalent

# Apply to real data column
data[[i]] <- unname(ruleset[[indexed]][mini_data])
data[[i]] <- unname(rules[[indexed]][mini_data])

}

Expand All @@ -192,6 +192,6 @@ lgb.prepare_rules2 <- function(data, rules = NULL) {

}

return(list(data = data, rules = ruleset))
return(list(data = data, rules = rules))

}

0 comments on commit 03ba730

Please sign in to comment.