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

Before Change

y_test = np.array(y_test)

i = Input(shape=(maxlen,))
x = Embedding(max_features, 128)(i)
x = TCN(nb_filters=64,
        kernel_size=6,
        dilations=[1, 2, 4, 8, 16, 32, 64])(x)
x = Dropout(0.5)(x)

After Change

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

Instances