未验证 提交 4cc9c1f2 编写于 作者: Q Qiao Longfei 提交者: GitHub

Merge pull request #1531 from fuhailin/Hailin

Update preprocess method for criteo to support python3
...@@ -51,7 +51,7 @@ class CategoryDictGenerator: ...@@ -51,7 +51,7 @@ class CategoryDictGenerator:
return res return res
def dicts_sizes(self): def dicts_sizes(self):
return map(len, self.dicts) return list(map(len, self.dicts))
class ContinuousFeatureGenerator: class ContinuousFeatureGenerator:
...@@ -61,8 +61,8 @@ class ContinuousFeatureGenerator: ...@@ -61,8 +61,8 @@ class ContinuousFeatureGenerator:
def __init__(self, num_feature): def __init__(self, num_feature):
self.num_feature = num_feature self.num_feature = num_feature
self.min = [sys.maxint] * num_feature self.min = [sys.maxsize] * num_feature
self.max = [-sys.maxint] * num_feature self.max = [-sys.maxsize] * num_feature
def build(self, datafile, continous_features): def build(self, datafile, continous_features):
with open(datafile, 'r') as f: with open(datafile, 'r') as f:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册