哎,BUG满天飞,浪费我一个小时时间,绝绝子.

[htmars-hub]

class ContinuousFeatureGenerator:
"""
Clip continuous features.
"""

def __init__(self, num_feature):
    self.num_feature = num_feature

def build(self, datafile, continous_features):
    df = pd.read_csv(datafile, sep="\t", header=True)
    with open(datafile, 'r') as f:
        for line in f:
            features = line.rstrip('\n').split('\t')
            for i in range(0, self.num_feature):
                val = features[continous_features[i]]
                if val != '':
                    val = int(val)
                    if val > continous_clip[i]:
                        val = continous_clip[i] # 这个val弄了半天,也没存储,赋值,所以处理了有啥用呀

def gen(self, idx, val):
    if val == '':
        return 0.0
    val = float(val)
    return val