<reponame>UMDataScienceLab/research<gh_stars>10-100
# 2 indicates wide features of active party (with Y) and 1 indicates feature groups of passive party
group_ind_list = [2, 1, 1, 1, 1, 1, 1]
column_name_list = ['UserInfo_14',
                    'UserInfo_18',
                    'UserInfo_15',
                    'UserInfo_10',
                    'UserInfo_22',
                    'UserInfo_23',
                    'UserInfo_3',
                    'UserInfo_12',
                    'UserInfo_21',
                    'UserInfo_9',
                    'UserInfo_1',
                    'UserInfo_17',
                    'UserInfo_11',
                    'UserInfo_5',
                    'UserInfo_13',
                    'UserInfo_6',
                    'UserInfo_16',
                    'UserInfo_20_Longitude',
                    'UserInfo_20_Latitude',
                    'UserInfo_20_LongpLat',
                    'UserInfo_20_LongmLat',
                    'UserInfo_20_CityRank',
                    'UserInfo_CityRank_median',
                    'UserInfo_Latitude_median',
                    'UserInfo_Longitude_median',
                    'UserInfo_LongmLat_median',
                    'UserInfo_LongpLat_median',
                    'UserInfo_CityRank_std',
                    'UserInfo_Latitude_std',
                    'UserInfo_Longitude_std',
                    'UserInfo_LongmLat_std',
                    'UserInfo_LongpLat_std',
                    'ThirdParty_Info_1_median',
                    'ThirdParty_Info_10_median',
                    'ThirdParty_Info_11_median',
                    'ThirdParty_Info_12_median',
                    'ThirdParty_Info_13_median',
                    'ThirdParty_Info_14_median',
                    'ThirdParty_Info_15_median',
                    'ThirdParty_Info_16_median',
                    'ThirdParty_Info_17_median',
                    'ThirdParty_Info_2_median',
                    'ThirdParty_Info_3_median',
                    'ThirdParty_Info_4_median',
                    'ThirdParty_Info_5_median',
                    'ThirdParty_Info_6_median',
                    'ThirdParty_Info_7_median',
                    'ThirdParty_Info_8_median',
                    'ThirdParty_Info_9_median',
                    'ThirdParty_Info_1_std',
                    'ThirdParty_Info_10_std',
                    'ThirdParty_Info_11_std',
                    'ThirdParty_Info_12_std',
                    'ThirdParty_Info_13_std',
                    'ThirdParty_Info_14_std',
                    'ThirdParty_Info_15_std',
                    'ThirdParty_Info_16_std',
                    'ThirdParty_Info_17_std',
                    'ThirdParty_Info_2_std',
                    'ThirdParty_Info_3_std',
                    'ThirdParty_Info_4_std',
                    'ThirdParty_Info_5_std',
                    'ThirdParty_Info_6_std',
                    'ThirdParty_Info_7_std',
                    'ThirdParty_Info_8_std',
                    'ThirdParty_Info_9_std',
                    'ThirdParty_Info_1_min',
                    'ThirdParty_Info_10_min',
                    'ThirdParty_Info_11_min',
                    'ThirdParty_Info_12_min',
                    'ThirdParty_Info_13_min',
                    'ThirdParty_Info_14_min',
                    'ThirdParty_Info_15_min',
                    'ThirdParty_Info_16_min',
                    'ThirdParty_Info_17_min',
                    'ThirdParty_Info_2_min',
                    'ThirdParty_Info_3_min',
                    'ThirdParty_Info_4_min',
                    'ThirdParty_Info_5_min',
                    'ThirdParty_Info_6_min',
                    'ThirdParty_Info_7_min',
                    'ThirdParty_Info_8_min',
                    'ThirdParty_Info_9_min',
                    'ThirdParty_Info_1_max',
                    'ThirdParty_Info_10_max',
                    'ThirdParty_Info_11_max',
                    'ThirdParty_Info_12_max',
                    'ThirdParty_Info_13_max',
                    'ThirdParty_Info_14_max',
                    'ThirdParty_Info_15_max',
                    'ThirdParty_Info_16_max',
                    'ThirdParty_Info_17_max',
                    'ThirdParty_Info_2_max',
                    'ThirdParty_Info_3_max',
                    'ThirdParty_Info_4_max',
                    'ThirdParty_Info_5_max',
                    'ThirdParty_Info_6_max',
                    'ThirdParty_Info_7_max',
                    'ThirdParty_Info_8_max',
                    'ThirdParty_Info_9_max',
                    'ThirdParty_Info_1_first',
                    'ThirdParty_Info_10_first',
                    'ThirdParty_Info_11_first',
                    'ThirdParty_Info_12_first',
                    'ThirdParty_Info_13_first',
                    'ThirdParty_Info_14_first',
                    'ThirdParty_Info_15_first',
                    'ThirdParty_Info_16_first',
                    'ThirdParty_Info_17_first',
                    'ThirdParty_Info_2_first',
                    'ThirdParty_Info_3_first',
                    'ThirdParty_Info_4_first',
                    'ThirdParty_Info_5_first',
                    'ThirdParty_Info_6_first',
                    'ThirdParty_Info_7_first',
                    'ThirdParty_Info_8_first',
                    'ThirdParty_Info_9_first',
                    'Education_Info8',
                    'Education_Info4',
                    'Education_Info7',
                    'Education_Info2',
                    'Education_Info5',
                    'Education_Info3',
                    'Education_Info1',
                    'Education_Info6',
                    'SocialNetwork_16',
                    'SocialNetwork_10',
                    'SocialNetwork_3',
                    'SocialNetwork_13',
                    'SocialNetwork_4',
                    'SocialNetwork_9',
                    'SocialNetwork_17',
                    'SocialNetwork_8',
                    'SocialNetwork_6',
                    'SocialNetwork_14',
                    'SocialNetwork_5',
                    'SocialNetwork_15',
                    'SocialNetwork_2',
                    'SocialNetwork_12',
                    'SocialNetwork_7',
                    'WeblogInfo_48',
                    'WeblogInfo_17',
                    'WeblogInfo_7',
                    'WeblogInfo_25',
                    'WeblogInfo_3',
                    'WeblogInfo_29',
                    'WeblogInfo_8',
                    'WeblogInfo_33',
                    'WeblogInfo_34',
                    'WeblogInfo_2',
                    'WeblogInfo_35',
                    'WeblogInfo_28',
                    'WeblogInfo_15',
                    'WeblogInfo_38',
                    'WeblogInfo_30',
                    'WeblogInfo_39',
                    'WeblogInfo_36',
                    'WeblogInfo_9',
                    'WeblogInfo_5',
                    'WeblogInfo_24',
                    'WeblogInfo_56',
                    'WeblogInfo_16',
                    'WeblogInfo_27',
                    'WeblogInfo_4',
                    'WeblogInfo_26',
                    'WeblogInfo_6',
                    'WeblogInfo_18',
                    'WeblogInfo_57',
                    'WeblogInfo_42',
                    'WeblogInfo_14',
                    'WeblogInfo_21',
                    'WeblogInfo_20',
                    'WeblogInfo_19']

group_info = [(0, (6, False)),
              # (6, (11, True)),
              (17, (15, False)),
              (32, (85, False)),
              (117, (8, True)),
              (125, (12, False), (3, True)),
              (140, (30, False), (3, True))]

embedding_shape_map = {'UserInfo_13': (3, 2),
                       'UserInfo_12': (3, 2),
                       # 'UserInfo_22': (9, 9),
                       'UserInfo_17': (2, 1),
                       'UserInfo_21': (2, 1),
                       'UserInfo_5': (3, 2),
                       'UserInfo_3': (9, 6),
                       'UserInfo_11': (3, 2),
                       'UserInfo_16': (6, 4),
                       'UserInfo_1': (9, 6),
                       'UserInfo_6': (3, 2),
                       # 'UserInfo_23': (31, 15),
                       'UserInfo_9': (4, 3),
                       'Education_Info3': (3, 2),
                       'Education_Info5': (2, 2),
                       'Education_Info1': (2, 2),
                       'Education_Info7': (2, 2),
                       'Education_Info6': (6, 5),
                       'Education_Info8': (7, 6),
                       'Education_Info2': (7, 6),
                       'Education_Info4': (6, 5),
                       'SocialNetwork_2': (3, 2),
                       'SocialNetwork_7': (3, 2),
                       'SocialNetwork_12': (3, 2),
                       'WeblogInfo_19': (8, 6),
                       'WeblogInfo_21': (5, 4),
                       'WeblogInfo_20': (45, 15)}

# 2 indicates wide features of active party (with Y) and 1 indicates feature groups of passive party
group_ind_list = [2, 1, 1, 1, 1, 1, 1]
column_name_list = ['UserInfo_14',
                    'UserInfo_18',
                    'UserInfo_15',
                    'UserInfo_10',
                    'UserInfo_22',
                    'UserInfo_23',
                    'UserInfo_3',
                    'UserInfo_12',
                    'UserInfo_21',
                    'UserInfo_9',
                    'UserInfo_1',
                    'UserInfo_17',
                    'UserInfo_11',
                    'UserInfo_5',
                    'UserInfo_13',
                    'UserInfo_6',
                    'UserInfo_16',
                    'UserInfo_20_Longitude',
                    'UserInfo_20_Latitude',
                    'UserInfo_20_LongpLat',
                    'UserInfo_20_LongmLat',
                    'UserInfo_20_CityRank',
                    'UserInfo_CityRank_median',
                    'UserInfo_Latitude_median',
                    'UserInfo_Longitude_median',
                    'UserInfo_LongmLat_median',
                    'UserInfo_LongpLat_median',
                    'UserInfo_CityRank_std',
                    'UserInfo_Latitude_std',
                    'UserInfo_Longitude_std',
                    'UserInfo_LongmLat_std',
                    'UserInfo_LongpLat_std',
                    'ThirdParty_Info_1_median',
                    'ThirdParty_Info_10_median',
                    'ThirdParty_Info_11_median',
                    'ThirdParty_Info_12_median',
                    'ThirdParty_Info_13_median',
                    'ThirdParty_Info_14_median',
                    'ThirdParty_Info_15_median',
                    'ThirdParty_Info_16_median',
                    'ThirdParty_Info_17_median',
                    'ThirdParty_Info_2_median',
                    'ThirdParty_Info_3_median',
                    'ThirdParty_Info_4_median',
                    'ThirdParty_Info_5_median',
                    'ThirdParty_Info_6_median',
                    'ThirdParty_Info_7_median',
                    'ThirdParty_Info_8_median',
                    'ThirdParty_Info_9_median',
                    'ThirdParty_Info_1_std',
                    'ThirdParty_Info_10_std',
                    'ThirdParty_Info_11_std',
                    'ThirdParty_Info_12_std',
                    'ThirdParty_Info_13_std',
                    'ThirdParty_Info_14_std',
                    'ThirdParty_Info_15_std',
                    'ThirdParty_Info_16_std',
                    'ThirdParty_Info_17_std',
                    'ThirdParty_Info_2_std',
                    'ThirdParty_Info_3_std',
                    'ThirdParty_Info_4_std',
                    'ThirdParty_Info_5_std',
                    'ThirdParty_Info_6_std',
                    'ThirdParty_Info_7_std',
                    'ThirdParty_Info_8_std',
                    'ThirdParty_Info_9_std',
                    'ThirdParty_Info_1_min',
                    'ThirdParty_Info_10_min',
                    'ThirdParty_Info_11_min',
                    'ThirdParty_Info_12_min',
                    'ThirdParty_Info_13_min',
                    'ThirdParty_Info_14_min',
                    'ThirdParty_Info_15_min',
                    'ThirdParty_Info_16_min',
                    'ThirdParty_Info_17_min',
                    'ThirdParty_Info_2_min',
                    'ThirdParty_Info_3_min',
                    'ThirdParty_Info_4_min',
                    'ThirdParty_Info_5_min',
                    'ThirdParty_Info_6_min',
                    'ThirdParty_Info_7_min',
                    'ThirdParty_Info_8_min',
                    'ThirdParty_Info_9_min',
                    'ThirdParty_Info_1_max',
                    'ThirdParty_Info_10_max',
                    'ThirdParty_Info_11_max',
                    'ThirdParty_Info_12_max',
                    'ThirdParty_Info_13_max',
                    'ThirdParty_Info_14_max',
                    'ThirdParty_Info_15_max',
                    'ThirdParty_Info_16_max',
                    'ThirdParty_Info_17_max',
                    'ThirdParty_Info_2_max',
                    'ThirdParty_Info_3_max',
                    'ThirdParty_Info_4_max',
                    'ThirdParty_Info_5_max',
                    'ThirdParty_Info_6_max',
                    'ThirdParty_Info_7_max',
                    'ThirdParty_Info_8_max',
                    'ThirdParty_Info_9_max',
                    'ThirdParty_Info_1_first',
                    'ThirdParty_Info_10_first',
                    'ThirdParty_Info_11_first',
                    'ThirdParty_Info_12_first',
                    'ThirdParty_Info_13_first',
                    'ThirdParty_Info_14_first',
                    'ThirdParty_Info_15_first',
                    'ThirdParty_Info_16_first',
                    'ThirdParty_Info_17_first',
                    'ThirdParty_Info_2_first',
                    'ThirdParty_Info_3_first',
                    'ThirdParty_Info_4_first',
                    'ThirdParty_Info_5_first',
                    'ThirdParty_Info_6_first',
                    'ThirdParty_Info_7_first',
                    'ThirdParty_Info_8_first',
                    'ThirdParty_Info_9_first',
                    'Education_Info8',
                    'Education_Info4',
                    'Education_Info7',
                    'Education_Info2',
                    'Education_Info5',
                    'Education_Info3',
                    'Education_Info1',
                    'Education_Info6',
                    'SocialNetwork_16',
                    'SocialNetwork_10',
                    'SocialNetwork_3',
                    'SocialNetwork_13',
                    'SocialNetwork_4',
                    'SocialNetwork_9',
                    'SocialNetwork_17',
                    'SocialNetwork_8',
                    'SocialNetwork_6',
                    'SocialNetwork_14',
                    'SocialNetwork_5',
                    'SocialNetwork_15',
                    'SocialNetwork_2',
                    'SocialNetwork_12',
                    'SocialNetwork_7',
                    'WeblogInfo_48',
                    'WeblogInfo_17',
                    'WeblogInfo_7',
                    'WeblogInfo_25',
                    'WeblogInfo_3',
                    'WeblogInfo_29',
                    'WeblogInfo_8',
                    'WeblogInfo_33',
                    'WeblogInfo_34',
                    'WeblogInfo_2',
                    'WeblogInfo_35',
                    'WeblogInfo_28',
                    'WeblogInfo_15',
                    'WeblogInfo_38',
                    'WeblogInfo_30',
                    'WeblogInfo_39',
                    'WeblogInfo_36',
                    'WeblogInfo_9',
                    'WeblogInfo_5',
                    'WeblogInfo_24',
                    'WeblogInfo_56',
                    'WeblogInfo_16',
                    'WeblogInfo_27',
                    'WeblogInfo_4',
                    'WeblogInfo_26',
                    'WeblogInfo_6',
                    'WeblogInfo_18',
                    'WeblogInfo_57',
                    'WeblogInfo_42',
                    'WeblogInfo_14',
                    'WeblogInfo_21',
                    'WeblogInfo_20',
                    'WeblogInfo_19']

group_info = [(0, (6, False)),
              # (6, (11, True)),
              (17, (15, False)),
              (32, (85, False)),
              (117, (8, True)),
              (125, (12, False), (3, True)),
              (140, (30, False), (3, True))]

embedding_shape_map = {'UserInfo_13': (3, 2),
                       'UserInfo_12': (3, 2),
                       # 'UserInfo_22': (9, 9),
                       'UserInfo_17': (2, 1),
                       'UserInfo_21': (2, 1),
                       'UserInfo_5': (3, 2),
                       'UserInfo_3': (9, 6),
                       'UserInfo_11': (3, 2),
                       'UserInfo_16': (6, 4),
                       'UserInfo_1': (9, 6),
                       'UserInfo_6': (3, 2),
                       # 'UserInfo_23': (31, 15),
                       'UserInfo_9': (4, 3),
                       'Education_Info3': (3, 2),
                       'Education_Info5': (2, 2),
                       'Education_Info1': (2, 2),
                       'Education_Info7': (2, 2),
                       'Education_Info6': (6, 5),
                       'Education_Info8': (7, 6),
                       'Education_Info2': (7, 6),
                       'Education_Info4': (6, 5),
                       'SocialNetwork_2': (3, 2),
                       'SocialNetwork_7': (3, 2),
                       'SocialNetwork_12': (3, 2),
                       'WeblogInfo_19': (8, 6),
                       'WeblogInfo_21': (5, 4),
                       'WeblogInfo_20': (45, 15)}
# 2 indicates wide features of active party (with Y) and 1 indicates feature groups of passive party # (6, (11, True)), # 'UserInfo_22': (9, 9), # 'UserInfo_23': (31, 15),