Ejemplos de MLP en Python

Ejemplo n.º 1

def setup_tvae():
    enc = TVAEEncoder([2, 1], nonlinearity=nn.ReLU)
    dec = MLP([1, 2], initial_batchnorm=False, nonlinearity=nn.ReLU)
    return dt.decomposition.deep.TVAE(enc, dec, learning_rate=1e-3)

Ejemplo n.º 2

def test_mlp_sanity():
    mlp = MLP([100, 10, 2])
    with torch.no_grad():
        x = torch.empty((5, 100)).normal_()
        mlp(x)

Ejemplo n.º 3

def setup_tae():
    enc = MLP([2, 1], initial_batchnorm=False, nonlinearity=nn.Tanh)
    dec = MLP([1, 2], initial_batchnorm=False, nonlinearity=nn.Tanh)
    return dt.decomposition.deep.TAE(enc, dec, learning_rate=1e-3)