data_params = {
    'Math2': {
        'stu_num': 3911,
        'prob_num': 16,
        'know_num': 16,
        'batch_size': 32,
    },
    'Math1': {
        'stu_num': 4209,
        'prob_num': 15,
        'know_num': 11,
        'batch_size': 32
    },
    'junyi': {
        'stu_num': 10000,
        'prob_num': 734,
        'know_num': 734,
        'batch_size': 1024
    },
    'a910': {
        'stu_num': 4128,
        'prob_num': 17746,
        'know_num': 123,
        'batch_size': 128
    },
    'a17': {
        'stu_num': 1709,
        'prob_num': 3162,
        'know_num': 102,
        'batch_size': 128
    },
    'nips20':{
        'stu_num': 3000,
        'prob_num': 6000,
        'know_num': 268,
        'batch_size': 128
    },
    'a09':{
        'stu_num': 3514,
        'prob_num': 10000,
        'know_num': 95,
        'batch_size': 64
    }
}