Beispiel #1
0
def test_mr_embd():
    nb_words = 18000
    maxlen = 64
    embd_dim = 100
    X_train, Y_train, X_test, Y_test, nb_classes = load_mr(nb_words, maxlen, 'self')
    cnn1d_selfembd(X_train, Y_train, X_test, Y_test, nb_classes,
                   maxlen, nb_words, embd_dim,
                   100, 5, 32, 20, 'rmsprop')
Beispiel #2
0
def test_mr_w2v():
    maxlen = 64
    X_train, Y_train, X_test, Y_test, nb_classes = load_mr(0, maxlen, 'w2v')
    cnn1d_w2vembd(X_train, Y_train, X_test, Y_test, nb_classes,
                   maxlen,
                   100, 5, 32, 20, 'rmsprop')