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')
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')