def METHOD_NAME(self, is_infer=False):
    inputs = paddle.static.data(
        name="input", shape=[-1, self.feature_size], dtype="float32")
    label_income = paddle.static.data(
        name="label_income", shape=[-1, 1], dtype="int64", lod_level=0)
    label_marital = paddle.static.data(
        name="label_marital", shape=[-1, 1], dtype="int64", lod_level=0)
    if is_infer:
        return [inputs, label_income, label_marital]
    else:
        return [inputs, label_income, label_marital]