Exemple #1
0
def test_assemble_gpgru():
    for gp_type in ['GP', 'MSGP']:
        model = assemble('GP-GRU', [gru_configs['1H'], gp_configs[gp_type]])
        loss = [gen_gp_loss(gp) for gp in model.output_layers]
        model.compile(optimizer=optimizer, loss=loss)
        assert model.built
Exemple #2
0
def test_assemble_gru():
    for i in xrange(1, 3):
        model = assemble('GRU', gru_configs[str(i) + 'H'])
        model.compile(optimizer=optimizer, loss='mse')
        assert model.built
Exemple #3
0
def test_assemble_rnn():
    for i in xrange(1, 3):
        model = assemble('RNN', rnn_configs[str(i) + 'H'])
        model.compile(optimizer=optimizer, loss='mse')
        assert model.built
Exemple #4
0
def test_assemble_lstm():
    for i in xrange(1, 3):
        model = assemble('LSTM', lstm_configs[str(i) + 'H'])
        model.compile(optimizer=optimizer, loss='mse')
        assert model.built
Exemple #5
0
def test_assemble_narx():
    for i in xrange(3):
        model = assemble('NARX', narx_configs[str(i) + 'H'])
        model.compile(optimizer=optimizer, loss='mse')
        assert model.built