Ethan Rosenthal EthanRosenthal

gati / guess_candidate_model.py

Created November 2, 2016 15:42

Code for training an LSTM model for text classification using the keras library (Theano backend). Was used for guesscandidate.com.

	from sklearn.cross_validation import train_test_split
	from keras.preprocessing import sequence, text
	from keras.models import Sequential
	from keras.layers import (Dense, Dropout, Activation, Embedding, LSTM,
	Convolution1D, MaxPooling1D)

	# Embedding
	max_features = 20000
	maxlen = 100
	embedding_size = 32