Получите ошибку: TypeError: операция редукции 'argmax' не разрешена для этого dtype после выполнения кода
from sklearn.metrics import f1_score
#convert test label from one-hot to number
unique_label_dict = {unique_label[i]:i for i in range(0, len(unique_label))}
test_label = test_data.drop('input',axis=1).idxmax(axis=1).apply(lambda x:unique_label_dict[x])