jmquintana79
2/2/2017 - 9:27 AM

Mean F Score

Mean F Score

"""
Mean F Score

kaggle manual: https://www.kaggle.com/wiki/MeanFScore
scikit manual: http://scikit-learn.org/stable/modules/generated/sklearn.metrics.f1_score.html
"""

from sklearn.metrics import f1_score

y = [0,0,0,0,1,0,0,0,0,1,1,1,1] 
y_pred = [0,0,0,0,0,1,0,1,0,0,1,0,0] 
f1_score(y, y_pred, average='weighted') 
#0.50641025641025628