def gen_data_crosscat(mode, T): # edit transition list according to all_transitions = [] M_c = du.gen_M_c_from_T(T, cctypes=['continuous']*2) state = State.p_state(M_c, T) if mode == 'crp_mixture': # fix the views X_D = state.get_X_D(); X_L = state.get_X_L(); X_D = [X_D[0]] X_L['column_partition']['assignments'] = [1,1] state = State.p_state(M_c, T, X_L=X_L, X_D=X_D)
def gen_data_crosscat(mode, T): # edit transition list according to all_transitions = [] M_c = du.gen_M_c_from_T(T, cctypes=['continuous'] * 2) state = State.p_state(M_c, T) if mode == 'crp_mixture': # fix the views X_D = state.get_X_D() X_L = state.get_X_L() X_D = [X_D[0]] X_L['column_partition']['assignments'] = [1, 1] state = State.p_state(M_c, T, X_L=X_L, X_D=X_D)