Quiero calcular la precisión, la recuperación y la puntuación F1 para mi modelo binario KerasClassifier, pero no encuentro ninguna solución.
Aquí está mi 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])
Y luego pronostico nuevos datos de prueba y obtengo la matriz de confusión 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)
Pero, ¿hay alguna solución para obtener el puntaje de precisión, el puntaje F1, la precisión y la recuperación? (Si no es complicado, también el puntaje de validación cruzada, pero no es necesario para esta respuesta)
¡Gracias por cualquier ayuda!
Podrías usar el informe de clasificación scikit-learn . Para convertir sus etiquetas a un formato numérico o binario, eche un vistazo al codificador de etiquetas scikit-learn .
que le da (salida copiada del ejemplo scikit-learn):
fuente
También puedes probar como se menciona a continuación.
fuente
Pruebe esto: https://scikit-learn.org/stable/modules/generated/sklearn.metrics.precision_recall_fscore_support.html con Y_test, y_pred como parámetros.
fuente