machine-learning - Keras model.fit() - 使用哪种训练算法?

标签 machine-learning deep-learning keras theano



我在 Theano 之上使用 Keras 创建一个 MLP,我训练并用它来预测时间序列。独立于网络的结构和深度,我无法弄清楚(Keras 文档、StackOverflow、搜索网络...)哪种训练算法(反向传播,...)Keras 的 model.fit()函数正在使用。

在 Theano 中(之前没有使用 Keras)我可以定义自己调整参数的方式

self.train_step = theano.function(inputs=[u_in, t_in, lrate], outputs=[cost, y],
                                      on_unused_input='warn',
                                      updates=[(p, p - lrate * g) for p, g in zip(self.parameters, self.gradients)],
                                      allow_input_downcast=True)


找不到任何信息会让人担心我错过了一些重要的东西,这可能是一个完全愚蠢的问题。

有人可以帮我吗?预先非常感谢。

最佳答案

看例子here :

...
model = Sequential()
model.add(Conv2D(32, kernel_size=(3, 3),
                 activation='relu',
                 input_shape=input_shape))
model.add(Conv2D(64, (3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(0.25))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(num_classes, activation='softmax'))

model.compile(loss=keras.losses.categorical_crossentropy,
              optimizer=keras.optimizers.Adadelta(),
              metrics=['accuracy'])

model.fit(x_train, y_train,
          batch_size=batch_size,
          epochs=epochs,
          verbose=1,
          validation_data=(x_test, y_test))
score = model.evaluate(x_test, y_test, verbose=0)
...

model.fit 使用算法来预测结果,而是使用您描述的模型。然后在 model.compile 中指定优化器算法

例如

model.compile(loss=keras.losses.categorical_crossentropy,
          optimizer=**keras.optimizers.Adadelta()**,
          metrics=['accuracy'])

您可以在此处找到有关可用优化器的更多信息:https://keras.io/optimizers/

关于machine-learning - Keras model.fit() - 使用哪种训练算法?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45834191/

相关文章:

nlp - Keras LSTM 的内部运作

machine-learning - 我怎样才能升级我的caffe模型,这样每次使用它时就不会出现 'upgrade'

python - input_shape 和模型结构不匹配

opencv - 检测篮球篮球和球跟踪

python - 具有 Maxpooling1D 和 channel_first 的 Keras 模型

algorithm - 帮助解决版本空间(AI)问题

image - 将 Pytorch 的图像格式从 NHWC 更改为 NCHW

tensorflow - LSTM 的精度非常低

python - 如何在 Keras 中计算精度和召回率

machine-learning - 车杆的 SARSA 值近似值