def create_dataset(dataset=None, labels=None):
    """
    构造贷款数据集
    :return: 返回数据集和是否放贷(二类)
    """
    if not dataset:
        dataset = [[0, 0, 0, 0, 'no'],
                   [0, 0, 0, 1, 'no'],
                   [0, 1, 0, 1, 'yes'],
                   [0, 1, 1, 0, 'yes'],
                   [0, 0, 0, 0, 'no'],
                   [1, 0, 0, 0, 'no'],
                   [1, 0, 0, 1, 'no'],
                   [1, 1, 1, 1, 'yes'],
                   [1, 0, 1, 2, 'yes'],
                   [1, 0, 1, 2, 'yes'],
                   [2, 0, 1, 2, 'yes'],
                   [2, 0, 1, 1, 'yes'],
                   [2, 1, 0, 1, 'yes'],
                   [2, 1, 0, 2, 'yes'],
                   [2, 0, 0, 0, 'no']]

    if not labels:
        labels = ['不放贷', '放贷']
        # labels = ['年龄', '有工作', '有自己的房子', '信贷情况']

    return dataset, labels