def test_sg15():
nb_words = 8000
maxlen = 150
embd_dim = 100
X_train, Y_train, X_test, Y_test, nb_classes = load_sg15(nb_words, maxlen, 'self')
cnn1d_selfembd(X_train, Y_train, X_test, Y_test, nb_classes,
maxlen, nb_words, embd_dim,
100, 5, 50, 20, 'rmsprop')
def test_sg15_lstm():
nb_words = 10000 # for NNS speakers, should be sufficient
maxlen = 200
embd_dim = 100
X_train, Y_train, X_test, Y_test, nb_classes = load_sg15(nb_words, maxlen, 'self')
lstm_selfembd(X_train, Y_train, X_test, Y_test, nb_classes,
maxlen, nb_words, embd_dim,
50, 32, 20, 'rmsprop')
def test_sg15_w2v():
maxlen = 200
X_train, Y_train, X_test, Y_test, nb_classes = load_sg15(0, maxlen, 'w2v')
cnn1d_w2vembd(X_train, Y_train, X_test, Y_test, nb_classes,
maxlen,
100, 10, 64, 20, 'rmsprop')