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)
我已经搜索了一些错误消息,但还没找到我可以应用的东西.任何提示?
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?