Quero calcular a precisão, recall e pontuação F1 para o meu modelo binário KerasClassifier, mas não encontro nenhuma solução.
Aqui está o meu código real:
# Split dataset in train and test data
X_train, X_test, Y_train, Y_test = train_test_split(normalized_X, Y, test_size=0.3, random_state=seed)
# Build the model
model = Sequential()
model.add(Dense(23, input_dim=45, kernel_initializer='normal', activation='relu'))
model.add(Dense(1, kernel_initializer='normal', activation='sigmoid'))
# Compile model
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
tensorboard = TensorBoard(log_dir="logs/{}".format(time.time()))
time_callback = TimeHistory()
# Fit the model
history = model.fit(X_train, Y_train, validation_split=0.3, epochs=200, batch_size=5, verbose=1, callbacks=[tensorboard, time_callback])
E então estou prevendo novos dados de teste e obtendo a matriz de confusão como esta:
y_pred = model.predict(X_test)
y_pred =(y_pred>0.5)
list(y_pred)
cm = confusion_matrix(Y_test, y_pred)
print(cm)
Mas existe alguma solução para obter a pontuação de precisão, a pontuação de F1, a precisão e o recall? (Se não for complicado, também a pontuação da validação cruzada, mas não é necessária para esta resposta)
Obrigado por qualquer ajuda!
Você pode usar o relatório de classificação do scikit-learn . Para converter seus rótulos em um formato numérico ou binário, dê uma olhada no codificador de rótulos scikit-learn .
que fornece a você (saída copiada do exemplo do scikit-learn):
fonte
Você também pode tentar conforme mencionado abaixo.
fonte
Tente o seguinte: https://scikit-learn.org/stable/modules/generated/sklearn.metrics.precision_recall_fscore_support.html com Y_test, y_pred como parâmetros.
fonte