Skip to content

Commit 80781d2

Browse files
Merge pull request #372 from boneyag/dict-comprehension
Include dict comprehension
2 parents 80288a0 + 5086069 commit 80781d2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

category_encoders/leave_one_out.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def fit_column_map(self, series, y):
128128
codes[codes == -1] = len(categories)
129129
categories = np.append(categories, np.nan)
130130

131-
return_map = pd.Series(dict([(code, category) for code, category in enumerate(categories)]))
131+
return_map = pd.Series({code: category for code, category in enumerate(categories)})
132132

133133
result = y.groupby(codes).agg(['sum', 'count'])
134134
return result.rename(return_map)

0 commit comments

Comments
 (0)