from tensorflow.contrib.learn.python.learn.estimators import test_data

    bucketized_feature = feature_column.bucketized_column(
        cont_feature, test_data.get_quantile_based_buckets(iris.data, 10))
