Definition: F1-Score

Sei ein gelabelter Datensatz.
Sei ein (binärer) Klassifikator.

Als F1-Score definieren wir:

Der F1-Score ist also das harmonische Mittel von Präzision und Recall.

Anmerkung

Bei mehr als einer Klasse

Wenn wir ein Mehrklassen-Problem haben, berechnen wir F1, Precision, Recall, … jeweils pro Klasse.

Recall in scikit-learn

In Python erhalten wir den F1-Score eines Modells durch:

model = ...
X_test = ...
y_test = ...
 
from sklearn.metrics import f1_score
 
score = f1_score(y_test, model.predict(X_test))