TypeError: to_categorical() got an unexpected keyword argument 'nb_classes'

最新推荐文章于 2024-08-15 18:56:33 发布

转载最新推荐文章于 2024-08-15 18:56:33 发布 · 466 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/zeroingToOne/p/8484804.html

文章标签：

#人工智能

本文解决了一个在使用Keras的to_categorical()函数时出现的TypeError问题，具体错误为'to_categorical() got an unexpected keyword argument 'nb_classes'。文章提供了修改建议，将'nb_classes'改为'num_classes'。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在学习莫烦教程中keras教程时，报错：TypeError: to_categorical() got an unexpected keyword argument 'nb_classes'，代码如下

X_train = X_train.reshape(X_train.shape[0],-1) / 255 #normalize
X_test  = X_test.reshape(X_test.shape[0],-1) / 255 #normalize
y_train = np_utils.to_categorical(y_train, nb_classes=10)
y_test = np_utils.to_categorical(y_test, nb_classes=10)

错误原因：keras版本问题，将nb_classes改为num_classes
参考keras文档：http://keras-cn.readthedocs.io/en/latest/utils/

转载于:https://www.cnblogs.com/zeroingToOne/p/8484804.html