Добавить отсев в модель Keras - PullRequest
Новая
       80

Добавить отсев в модель Keras

0 голосов
/ 17 марта 2020

Я хотел бы добавить выпадение в скрытых слоях. Я не знаю, следует ли мне применять предложение после или до определения каждого скрытого слоя.

Может кто-нибудь мне помочь? 

def keras_model2(layers_dims):

    L = len(layers_dims)

    model = Sequential()
    model.add(Dense(layers_dims[1], input_shape=(layers_dims[0],), activation="relu"))

    for l in range(2, L-1):
        model.add(Dense(layers_dims[l], activation="relu", kernel_initializer="random_normal",
                bias_initializer="zeros"))
        model.add(Dropout(0.4))
    model.add(Dense(layers_dims[L-1], activation="softmax", kernel_initializer="random_normal",
                bias_initializer="zeros"))



    return model
...