Write a function that takes a confusion matrix as input and returns index of the class with the maximum f1 score.

def max_f1_score(cm):

"""

Inputs:

cm : confusion matrix, 2-d numpy array

Outputs:

integer, index of class with max f1 score

"""

### Write your code here