Я использовал этот код для своего проекта, но я хочу вычислить или изобразить ошибку каждого класса. У меня есть 6 классов. как я могу это сделать?
def plot_history(net_history):
history = network_history.history
losses = history['loss']
accuracies = history['acc']
plt.xlabel('Epochs')
plt.ylabel('loss')
plt.plot(losses)
plt.figure()
plt.xlabel('Epochs')
plt.ylabel('accuracy')
plt.plot(accuracies)
Создать мою модель
myinput = layers.Input(shape=(100,200))
conv1 = layers.Conv1D(16, 3, activation='relu', padding='same', strides=2)(myinput)
conv2 = layers.Conv1D(32, 3, activation='relu', padding='same', strides=2)(conv1)
flat = layers.Flatten()(conv2)
out_layer = layers.Dense(6, activation='softmax')(flat)
mymodel = Model(myinput, out_layer)
mymodel.summary()
mymodel.compile(optimizer=keras.optimizers.Adam(),
loss=keras.losses.categorical_crossentropy, metrics=['accuracy'])
тренируй мою модель
network_history = mymodel.fit(X_train, Y_train, batch_size=128,epochs=5, validation_split=0.2)
plot_history(network_history)
Оценка
test_loss, test_acc = mymodel.evaluate(X_test, Y_test)
test_labels_p = mymodel.predict(X_test)