如何设置Keras模型正确?

时间:2016-02-01 07:41:17

标签: python python-2.7 neural-network keras

我正在努力在Keras建造模型。我希望将所有功能都传递为self::node()[not(@includeclass) or @includeclass='yes']/ancestor::class/..... input=10。但是,一旦我运行下面的代码

output=3

我收到了一个错误:

def build_model(input_dim, output_classes):
    print 'Creating model ...', time.time() - start, mprof.memory_usage()
    model = Sequential()
    model.add(Dense(input_dim=input_dim, output_dim=8, activation=relu))
    model.add(Dropout(0.5))
    model.add(Dense(input_dim=8, output_dim=6, activation=relu))
    model.add(Dropout(0.5))
    model.add(Dense(output_dim=output_classes, activation='softmax'))
    model.compile(loss='categorical_crossentropy', optimizer='adadelta')

在我的情况下,模型初始化的正确参数是什么?

1 个答案:

答案 0 :(得分:0)

我复制了你的代码并且没有错误地执行它。(只是在我使用Python 3时评论了打印)

我设置input_dim = 10和output_classes = 3,没有问题。

可能是您正在为output_classes传递非整数值,如果是这种情况,则可能会出现一系列错误,具体取决于传递的对象类型。