d98df66f581dca38bbe7dfc299386921b65bda0d,solutionbox/code_free_ml/mltoolbox/code_free_ml/trainer/task.py,,make_prediction_output_tensors,#Any#Any#Any#Any#Any#,364

Before Change

    // if top_n == 0, this means use all the classes.
    if args.top_n == 0:
      top_n = probabilities.shape[1].value
    else:
      top_n = args.top_n

After Change

    // probabilities labels.
    if args.top_n == 0:
      predicted_index = tf.argmax(probabilities, axis=1)
      predicted = table.lookup(predicted_index)
      outputs.update({PG_CLASSIFICATION_FIRST_LABEL: predicted})
      probabilities_list = tf.unstack(probabilities, axis=1)
      for class_name, p in zip(class_names, probabilities_list):
        outputs[class_name] = p

Instances