python数据分析与挖掘实战笔记二:第99页神经网络训练出现的错误'Some keys in session_kwargs are not supported at this time: %s'

在使用神经网络模型预测销量高低时,系统指出模型训练时出现错误:

ValueError Traceback (most recent call last)
<ipython-input-20-e46e29b76a5e> in <module>()
----> 1 model.fit(x, y, epochs=10, batch_size=10) #训练模型

~/anaconda3/lib/python3.6/site-packages/keras/engine/training.py in fit(self, x, y, batch_size, epochs, verbose, callbacks, validation_split, validation_data, shuffle, class_weight, sample_weight, initial_epoch, steps_per_epoch, validation_steps, **kwargs)
   1006         else:
   1007             ins = x + y + sample_weights
-> 1008         self._make_train_function()
   1009         f = self.train_function
   1010 
ValueError: ('Some keys in session_kwargs are not supported at this time: %s', dict_keys(['class_mode']))

实际上出现错误的地方是在模型编译处,错误提示后面具体指出了是在self._make_train_function()当中,说明是模型的某个训练方法出现了问题,而模型训练代码中没有关于模型训练的方法的参数,而是在模型的编译上。查看keras官方文档可以知道编译方法compile()方法只有三个参数,分别是优化器optimizer,损失函数loss和指标列表metrics,没有class_mode这个参数,对于分类问题,用metrics设定为metrics=[‘accuracy’]即可。
所以要将模型编译处的代码修改为以下内容:

model.compile(loss='binary_crossentropy', optimizer='adam', class_metrics=['accuracy'])
阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭