color_names.py 文件源码

def build_color2words_model(max_tokens, dim):
    """Build a model that learns to generate words from colors.

    :param max_tokens:
    :param dim:
    :return:
    """
    model = Sequential()
    model.add(Masking(mask_value = -1, input_shape = (max_tokens, 3)))
    model.add(LSTM(256, return_sequences = True))
    model.add(TimeDistributed(Dense(dim)))

    model.compile(loss = "mse", optimizer = "adam")
    return model