Skip to content

Instantly share code, notes, and snippets.

@rajeshmr

rajeshmr/product-titles-classifiy-text-cnn.py Secret

Created December 7, 2017 10:34
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save rajeshmr/aab648b10e2416cfa7e65d13272065ab to your computer and use it in GitHub Desktop.
Save rajeshmr/aab648b10e2416cfa7e65d13272065ab to your computer and use it in GitHub Desktop.
Download ZIP
Raw
product-titles-classifiy-text-cnn.py
model_1 = Sequential()
model_1.add(embedding_layer)
model_1.add(Conv1D(250,3,padding='valid',activation='relu',strides=1))
model_1.add(GlobalMaxPooling1D())
model_1.add(Dense(250))
model_1.add(Dropout(0.2))
model_1.add(Activation('relu'))
model_1.add(Dense(3))
model_1.add(Activation('sigmoid'))
model_1.compile(loss='categorical_crossentropy',optimizer='rmsprop',metrics=['acc'])
model_1.summary()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment