【发布时间】:2021-08-18 17:01:31
【问题描述】:
我正在尝试按照本指南 https://machinelearningmastery.com/how-to-calculate-precision-recall-f1-and-more-for-deep-learning-models/ 查找模型性能指标(F1 分数、准确度、召回率)
这个确切的代码几个月前还在工作,但现在返回各种错误,非常令人困惑,因为我没有更改此代码的一个字符。也许包更新改变了一些事情?
我使用 model.fit 拟合序列模型,然后使用 model.evaluate 来查找测试准确度。现在我正在尝试使用 model.predict_classes 进行类预测(模型是一个多类分类器)。代码如下:
model = Sequential()
model.add(Dense(24, input_dim=13, activation='relu'))
model.add(Dense(18, activation='relu'))
model.add(Dense(6, activation='softmax'))
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
-
history = model.fit(X_train, y_train, batch_size = 256, epochs = 10, verbose = 2, validation_split = 0.2)
-
score, acc = model.evaluate(X_test, y_test,verbose=2, batch_size= 256)
print('test accuracy:', acc)
-
yhat_classes = model.predict_classes(X_test)
最后一行返回错误“AttributeError: 'Sequential' object has no attribute 'predict_classes'”
这个确切的代码不久前还在工作,所以有点挣扎,感谢您的帮助
【问题讨论】:
-
你是在 colab 上运行这个吗?