def preprocessing_fn(inputs): weights = inputs.get('weights', None) x_bucketized = tft.bucketize_per_key(inputs['x'], inputs['key'], num_buckets=3, epsilon=0.00001, weights=weights) return {'x_bucketized': x_bucketized}
def preprocessing_fn(inputs): x_bucketized = tft.bucketize_per_key(inputs['x'], inputs['key'], num_buckets=3, epsilon=0.00001) return {'x_bucketized': x_bucketized}