NN.py 文件源码

def __build_keras_model(self):
        models = []

        model_artist_id = Sequential()
        model_artist_id.add(Embedding(100, 10, input_length=1))
        model_artist_id.add(Reshape(target_shape=(10,)))
        models.append(model_artist_id)

        model_week = Sequential()
        model_week.add(Embedding(7, 2, input_length=1))
        model_week.add(Reshape(target_shape=(6,)))
        models.append(model_week)

#         model_gender = Sequential()
#         model_gender.add(Embedding(1, 3, input_length=1))
#         model_gender.add(Reshape(target_shape=(3,)))
#         models.append(model_gender)

        model_day = Sequential()
        model_day.add(Embedding(1, 10, input_length=1))
        model_day.add(Reshape(target_shape=(10,)))
        models.append(model_day)

#         model_language = Sequential()
#         model_language.add(Embedding(1, 3, input_length=1))
#         model_language.add(Reshape(target_shape=(3,)))
#         models.append(model_language)

        model_others = Sequential()
        model_others.add(Reshape((self.others_dim,), input_shape=(self.others_dim,)))
        models.append(model_others)

        self.model = Sequential()
        self.model.add(Merge(models, mode='concat'))
        self.model.add(Dense(100, init='uniform'))
        self.model.add(Activation('relu'))
        self.model.add(Dense(200, init='uniform'))
        self.model.add(Activation('relu'))
        self.model.add(Dense(1))

        self.model.compile(loss='mean_absolute_error', optimizer='adam')