我试图使用代码片段训练用 rasa 构建的聊天机器人
agent.train(data,augmentation_factor=50,
epochs=500,
batch_size=10)
并得到以下错误。我知道它不是 rasa_core 错误,而是可能与 keras 相关的错误。
rasa_core/policies/keras_policy.py", line 177, in train
**params)
TypeError: fit() got multiple values for keyword argument 'batch_size'
我是机器人构建的新手,之前从未在深度学习项目中做过。使用以下依赖项 rasa_core==0.12.x keras==2.1.6
我遇到了同样的错误,经过多次搜索,我确实找到了解决方案。这不是最好的解决方案,但它可以提供帮助!
转到第 172-177 行中的 rasa_core\rasa_core\policies\keras_policy.py 并从 model.fit 中删除 epochs 和 batch_size 参数。改变这个:
params = self._get_valid_params(self.model.fit, **kwargs)
self.model.fit(shuffled_X, shuffled_y,
epochs=self.epochs, batch_size=self.batch_size,
**params)
对此:
params = self._get_valid_params(self.model.fit, **kwargs)
self.model.fit(shuffled_X, shuffled_y,**params)
然后你可以在 agent.train() 中传递 epochos 和 batch_size 参数
我希望它会有所帮助!
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句