565960370df285169f4a2b2a7368e3d9f07e6617,tasks/imdb_tcn.py,,,#,15

Before Change

x = TCN(nb_filters=64,
        kernel_size=6,
        dilations=[1, 2, 4, 8, 16, 32, 64])(x)
x = Dropout(0.5)(x)
x = Dense(1, activation="sigmoid")(x)

model = Model(inputs=[i], outputs=[x])

model.summary()

After Change

y_train = np.array(y_train)
y_test = np.array(y_test)

model = Sequential([
    Embedding(max_features, 128, input_shape=(maxlen,)),
    TCN(kernel_size=6, dilations=[1, 2, 4, 8, 16]),
    Dense(1, activation="sigmoid")
])

print(f"TCN receptive field: {model.layers[1].receptive_field}.")

model.summary()

Instances