Exemplo n.º 1
0
def nnet():
    model = df.Sequential()
    model.add(df.AddConstant(1.0))
    model.add(df.Log())
    model.add(df.BatchNormalization(93))
    model.add(df.Dropout(0.1))
    model.add(df.Linear(93, 512))
    model.add(df.BatchNormalization(512))
    model.add(df.ReLU())
    model.add(df.Dropout(0.5))

    model.add(df.Linear(512, 512))
    model.add(df.BatchNormalization(512))
    model.add(df.ReLU())
    model.add(df.Dropout(0.5))

    model.add(df.Linear(512, 512))
    model.add(df.BatchNormalization(512))
    model.add(df.ReLU())
    model.add(df.Dropout(0.5))

    model.add(df.Linear(512, 9))
    model.add(df.SoftMax())
    return model