How can I use a different value of k in top_k_categorical_accuracy? Is is possible to change a parameter somewhere?