Keras: model.fit() 和 model.fit_generator() 返回历史对象。我如何获取 Keras 模型?

我正在进行一个指导性的 RNN 项目。我使用教科书作为指导,但很多事情都是自己做的。我遇到了一个问题,就是下面的 history 不是一个 Keras 模型,而是一个历史对象。

from keras.models import Sequentialfrom keras import layersfrom keras.optimizers import RMSpropfrom keras.layers import LSTMmodel = Sequential()model.add(layers.Flatten(input_shape=(7,data.shape[-1])))model.add(layers.Dense(32,activation='relu'))model.add(layers.Dense(1))val_steps = 99999//20model.compile(optimizer=RMSprop(),loss='mae')history = model.fit_generator(trainGen,                             steps_per_epoch=250,                             epochs=20,                             validation_data=valGen,                             validation_steps=val_steps,                             use_multiprocessing=False)

当我输入下面的代码时,由于 history 是一个 History 对象,所以出现了错误。有没有办法提取一个 Keras 对象?提前感谢您。

predictions = history.predict(testData)

回答:

抱歉我还不能评论。为什么你在 history 上调用 predict,而不是在模型本身调用呢?

predictions = model.predict(testData)

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注