def make_color_data(data_dict):
    '\n    make a dataset consisting of the i-band mag and the five colors\n    Returns:\n    --------\n    input_data: (nd-array)\n    array of imag and 5 colors\n    '
    input_data = data_dict['mag_i_lsst']
    bands = ['u', 'g', 'r', 'i', 'z', 'y']
    for i in range(5):
        band1 = data_dict[f'mag_{bands[i]}_lsst']
        band2 = data_dict[f'mag_{bands[(i + 1)]}_lsst']
        input_data = np.vstack((input_data, (band1 - band2)))
    return input_data.T
make a dataset consisting of the i-band mag and the five colors
Returns:
--------
input_data: (nd-array)
array of imag and 5 colors
def make_color_data(data_dict):
    '\n    make a dataset consisting of the i-band mag and the five colors\n    Returns:\n    --------\n    input_data: (nd-array)\n    array of imag and 5 colors\n    '
    input_data = data_dict['mag_i_lsst']
    bands = ['u', 'g', 'r', 'i', 'z', 'y']
    for i in range(5):
        band1 = data_dict[f'mag_{bands[i]}_lsst']
        band2 = data_dict[f'mag_{bands[(i + 1)]}_lsst']
        input_data = np.vstack((input_data, (band1 - band2)))
    return input_data.T