model.py 文件源码

def conv_unit(inp, n_gram, no_word = 200, window = 2):
    out = Conv1D(no_word, window, strides=1, padding="valid", activation='relu')(inp)
    out = TimeDistributed(Dense(5, input_shape=(n_gram, no_word)))(out)
    out = ZeroPadding1D(padding=(0, window-1))(out)
    return out