相关疑难解决方法(0)

混淆矩阵不支持多标签指示符

multilabel-indicator is not supported 是我尝试运行时收到的错误消息:

confusion_matrix(y_test, predictions)

y_test是一个DataFrame形状:

Horse | Dog | Cat
1       0     0
0       1     0
0       1     0
...     ...   ...
Run Code Online (Sandbox Code Playgroud)

predictions是一个numpy array:

[[1, 0, 0],
 [0, 1, 0],
 [0, 1, 0]]
Run Code Online (Sandbox Code Playgroud)

我已经搜索了一些错误消息,但还没找到我可以应用的东西.任何提示?

python numpy classification scikit-learn

17
推荐指数
4
解决办法
1万
查看次数

Get confusion matrix from a Keras model

I have the following NN model using Keras:

import numpy as np
from keras import Sequential
from keras.layers import Dense

path = 'pima-indians-diabetes.data.csv'
dataset = np.loadtxt(path, delimiter=",")
X = dataset[:,0:8]
Y = dataset[:,8]
X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=0.2)

model = Sequential()
model.add(Dense(16, input_dim=8, activation='relu'))
model.add(Dense(32, activation='relu'))
model.add(Dense(1, activation='sigmoid'))

model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
model.fit(X_train, y_train, epochs=100, batch_size=16, validation_data=(X_test, y_test))
Run Code Online (Sandbox Code Playgroud)

Kindly, is it possible to extract the confusion matrix? How?

keras

0
推荐指数
1
解决办法
61
查看次数

标签 统计

classification ×1

keras ×1

numpy ×1

python ×1

scikit-learn ×1