原代码如下:
def define_model(vocad_size,max_length):
model=Sequential()
model.add(Embedding(vocad_size,100,input_length=max_length))
model.add(Conv1D(filters=32,kernel_size=5,activation='relu'))
model.add(MaxPooling1D(pool_size=2))
model.add(Flatten())
model.add(Dense(16,activation='relu'))
model.add(Dense(8,activation='softmax'))
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
model.summary()
return model
#模型训练
def model_train():
model = define_model(vocad_size, max_length)
model.fit(X_train, Y_train, epochs=10, verbose=2)
model.save('词向量模型分类.h5')
#model_train()#
model_train()